run tests on cpu

Lightning-AI · Mar 22, 2024 · 98130f9 · 98130f9
1 parent 018cc89
commit 98130f9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_evaluate.py b/tests/test_evaluate.py
@@ -36,6 +36,7 @@ def test_run_eval(tmp_path, float_like):
         model_args=f"pretrained={checkpoint_path}",
         tasks=eval_tasks.split(","),
         limit=2,
+        device="cpu"
     )
 
     save_path = checkpoint_path/"results.json"