Fix test

SainsburyWellcomeCentre · Nov 5, 2024 · abaeb46 · abaeb46
1 parent af381bb
commit abaeb46
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_unit/test_track_video.py b/tests/test_unit/test_track_video.py
@@ -20,6 +20,8 @@ def mock_args():
         accelerator="gpu",
         annotations_file=None,
         save_video=None,
+        experiment_name=None,
+        mlflow_folder=Path(temp_dir) / "ml-runs",
     )