run test on cpu

Lightning-AI · Mar 29, 2024 · b3b693e · b3b693e
1 parent 4c712a2
commit b3b693e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_evaluate.py b/tests/test_evaluate.py
@@ -47,6 +47,7 @@ def test_evaluate_script(tmp_path, fake_checkpoint_dir, monkeypatch):
     fn_kwargs = dict(
         checkpoint_dir=fake_checkpoint_dir,
         out_dir=output_dir,
+        device="cpu"
     )
     stdout = StringIO()
     with redirect_stdout(stdout), mock.patch("sys.argv", ["evaluate.py"]):