diff --git a/.isort.cfg b/.isort.cfg index a3f2cb2..41ef440 100644 --- a/.isort.cfg +++ b/.isort.cfg @@ -6,4 +6,4 @@ ensure_newline_before_comments=True line_length=120 [settings] -known_third_party=wandb \ No newline at end of file +known_third_party=mujoco, wandb \ No newline at end of file diff --git a/argus/validate_real.py b/argus/validate_real.py index 808d6a1..43e626a 100644 --- a/argus/validate_real.py +++ b/argus/validate_real.py @@ -1,16 +1,14 @@ import os - from dataclasses import dataclass -from argus.utils import get_pose, xyzxyzw_to_xyzwxyz_SE3 from pathlib import Path import h5py import kornia import matplotlib.pyplot as plt +import mujoco import torch import tyro -import mujoco from argus import ROOT from argus.data import CameraCubePoseDatasetConfig from argus.models import NCameraCNN, NCameraCNNConfig diff --git a/scripts/mesh_conversion.py b/scripts/mesh_conversion.py index 59ec02c..e49466b 100644 --- a/scripts/mesh_conversion.py +++ b/scripts/mesh_conversion.py @@ -1,6 +1,8 @@ +from pathlib import Path + import trimesh + from argus import ROOT -from pathlib import Path if __name__ == "__main__": mesh_path = Path(ROOT) / "LeapProject/Assets/urdf/meshes" diff --git a/scripts/mujoco_rendering.py b/scripts/mujoco_rendering.py index 2bb8d5a..f21c3cc 100644 --- a/scripts/mujoco_rendering.py +++ b/scripts/mujoco_rendering.py @@ -1,11 +1,12 @@ -import mujoco +import json +from pathlib import Path + +import h5py import matplotlib.pyplot as plt +import mujoco import numpy as np from argus import ROOT -from pathlib import Path -import h5py -import json USE_JSON = False diff --git a/scripts/rotation_overfitting.py b/scripts/rotation_overfitting.py index ed96a73..8218921 100644 --- a/scripts/rotation_overfitting.py +++ b/scripts/rotation_overfitting.py @@ -1,5 +1,6 @@ -import torch +# flake8: noqa import pypose as pp +import torch import torchvision.models as models