From e2d4cd65ced95822c80da234841a8817b63bf0a9 Mon Sep 17 00:00:00 2001
From: sfmig <33267254+sfmig@users.noreply.github.com>
Date: Tue, 29 Oct 2024 16:26:41 +0000
Subject: [PATCH] Fix test on track video CLI

---
 tests/test_unit/test_track_video.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_unit/test_track_video.py b/tests/test_unit/test_track_video.py
index 6833242b..7d7ffa89 100644
--- 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,
     )