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

Signed-off-by: Sebastian Berg <[email protected]>
rapidsai · Nov 12, 2024 · 8c78e0c · 8c78e0c
1 parent cb663ec
commit 8c78e0c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/legateboost/test/test_examples.py b/legateboost/test/test_examples.py
@@ -65,7 +65,8 @@ 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,