TST: Try forcing gpus=0 in benchmark test subprocess call

Signed-off-by: Sebastian Berg <[email protected]>
rapidsai · Nov 12, 2024 · ddea6d7 · ddea6d7
1 parent cb663ec
commit ddea6d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/legateboost/test/test_examples.py b/legateboost/test/test_examples.py
@@ -65,7 +65,7 @@ def test_benchmark(benchmark_dir):
     del env["LEGATE_CONFIG"]
 
     subprocess.run(
-        "legate --cpus 2 legateboost_scaling.py --nrows 100 --ncols 5 --niter 2"
+        "legate --cpus=2 --gpus=0 legateboost_scaling.py --nrows 100 --ncols 5 --niter 2"
         + " --model_types tree,linear,krr,nn",
         shell=True,
         check=True,