reorganize imports after ruff

agimus-project · Nov 5, 2024 · 116b014 · 116b014
1 parent 7a1e96c
commit 116b014
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/happypose/pose_estimators/cosypose/cosypose/evaluation/evaluation.py b/happypose/pose_estimators/cosypose/cosypose/evaluation/evaluation.py
@@ -12,9 +12,16 @@
 from happypose.pose_estimators.cosypose.cosypose.evaluation.prediction_runner import (
     PredictionRunner,
 )
+from happypose.pose_estimators.cosypose.cosypose.integrated.detector import Detector
 from happypose.pose_estimators.cosypose.cosypose.integrated.pose_estimator import (
     PoseEstimator,
 )
+from happypose.pose_estimators.cosypose.cosypose.training.detector_models_cfg import (
+    check_update_config as check_update_config_detector,
+)
+from happypose.pose_estimators.cosypose.cosypose.training.detector_models_cfg import (
+    create_model_detector,
+)
 
 # Detection
 from happypose.pose_estimators.cosypose.cosypose.training.pose_models_cfg import (
@@ -42,20 +49,10 @@
 from happypose.toolbox.utils.distributed import get_rank, get_tmp_dir
 from happypose.toolbox.utils.logging import get_logger
 
-# """" Temporary imports
-
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 logger = get_logger(__name__)
 
-from happypose.pose_estimators.cosypose.cosypose.integrated.detector import Detector
-from happypose.pose_estimators.cosypose.cosypose.training.detector_models_cfg import (
-    check_update_config as check_update_config_detector,
-)
-from happypose.pose_estimators.cosypose.cosypose.training.detector_models_cfg import (
-    create_model_detector,
-)
-
 
 def load_detector(run_id, ds_name):
     run_dir = EXP_DIR / run_id