Fix test on track video CLI

SainsburyWellcomeCentre · Oct 29, 2024 · e2d4cd6 · e2d4cd6
1 parent 4efc06e
commit e2d4cd6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_unit/test_track_video.py b/tests/test_unit/test_track_video.py
@@ -17,8 +17,8 @@ def mock_args():
         video_path="/path/to/video.mp4",
         trained_model_path="/path/to/model.ckpt",
         output_dir=temp_dir,
-        device="cuda",
-        gt_path=None,
+        accelerator="gpu",
+        annotations_file=None,
         save_video=None,
     )