fixing sf tests for CI

edbeeching · Jul 23, 2023 · e6251f0 · e6251f0
1 parent 081d001
commit e6251f0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_sample_factory.py b/tests/test_sample_factory.py
@@ -10,6 +10,7 @@ def test_sample_factory_training():
     args.env_path = "examples/godot_rl_JumperHard/bin/JumperHard.x86_64"
     extras = []
     extras.append('--env=gdrl')
-    extras.extend(['--train_for_env_steps=1000'])
+    extras.append('--train_for_env_steps=1000')
+    extras.append('--device=cpu')
 
     sample_factory_training(args, extras)