[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
agimus-project · Mar 25, 2024 · 18ba8b4 · 18ba8b4
1 parent 294b529
commit 18ba8b4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/happypose/pose_estimators/cosypose/cosypose/scripts/run_inference_on_example.py b/happypose/pose_estimators/cosypose/cosypose/scripts/run_inference_on_example.py
@@ -85,7 +85,9 @@ def run_inference(
     rgb, depth, camera_data = load_observation_example(example_dir, load_depth=True)
     # TODO: cosypose forward does not work if depth is loaded detection
     # contrary to megapose
-    observation = ObservationTensor.from_numpy(rgb, depth=None, K=camera_data.K).to(device)
+    observation = ObservationTensor.from_numpy(rgb, depth=None, K=camera_data.K).to(
+        device
+    )
 
     # Load models
     pose_estimator = setup_pose_estimator(args.dataset, object_dataset)