linted all files

.isort.cfg

@@ -6,4 +6,4 @@ ensure_newline_before_comments=True
 line_length=120
 
 [settings]
-known_third_party=wandb
+known_third_party=mujoco, wandb

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

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"

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
 

scripts/rotation_overfitting.py

@@ -1,5 +1,6 @@
-import torch
+# flake8: noqa
 import pypose as pp
+import torch
 import torchvision.models as models