Remove vjepa from mock tests

mila-iqia · Sep 12, 2024 · d0adca7 · d0adca7
1 parent c169bf9
commit d0adca7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_mock.py b/tests/test_mock.py
@@ -14,7 +14,9 @@
     "llm-lora-ddp-nodes",
     "llm-lora-mp-gpus",
     "llm-full-mp-gpus",
-    "llm-full-mp-nodes"
+    "llm-full-mp-nodes",
+    "vjepa-single",
+    "vjepa-gpus"
 }
 
 def run_cli(*args, expected_code=0, msg=None):