From 18ba8b4cd9ffae30cf54ce0436e779a9de97db59 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 25 Mar 2024 16:18:32 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../cosypose/cosypose/scripts/run_inference_on_example.py | 4 +++- 1 file changed, 3 insertions(+), 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 index 7558bae7..c79f4b79 100644 --- 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)