Use --omps=0 and fix style

seberg · seberg · commit a208f76503e2 · 2025-07-02T08:13:08.000-07:00
Signed-off-by: Sebastian Berg &lt;sebastianb@nvidia.com&gt;
diff --git a/examples/batch_training/batch_training.png b/examples/batch_training/batch_training.png
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a831958b4bab9fafa697c1030fae7bd227e11d0865717c92ae4dead506adb32c
-size 42617
+oid sha256:813c8e961b73234764a632a41cf5ba1593e258f993e96f0abb95f812dc926879
+size 46632
diff --git a/examples/kernel_ridge_regression/kernel_ridge_regression.png b/examples/kernel_ridge_regression/kernel_ridge_regression.png
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:be8e9ffe773df62be535efbac096f6a46f751247f3b9c31b2e4e469fcb0554ca
-size 67018
+oid sha256:b512196a4b8f348774b401a86198b7ec4ee1063dba8bf13f725b75f5d2b5082f
+size 46747
diff --git a/examples/linear_model/linear_model.png b/examples/linear_model/linear_model.png
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c78bb070c7c49818d01c2fca0e6b7520124ecdc230e938ccf581e84a7804086a
-size 96563
+oid sha256:987f3f514232b14a577dc9d045d2f3537e483e79b1d68abba60d2176c0b1fc5c
+size 93933
diff --git a/examples/probabalistic_regression/probabilistic_regression.gif b/examples/probabalistic_regression/probabilistic_regression.gif
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:03eb594a5ea974f86f938c08a78bc9fe0e07e3d70e7bfde3eaa51076a0e26730
-size 848444
+oid sha256:929d12d762d766fed9cd2a364e9d3280bc7cd42bf6ff78a51b5fdc1e93ba9a91
+size 134091
diff --git a/legateboost/test/test_examples.py b/legateboost/test/test_examples.py
@@ -65,14 +65,15 @@ def test_benchmark(benchmark_dir):
     env = os.environ.copy()
     del env["LEGATE_CONFIG"]
 
-    cmd = ("legate --cpus=2 --gpus=0"
-           " legateboost_scaling.py --nrows 100 --ncols 5 --niter 2"
-           " --model_types tree,linear,krr,nn")
+    cmd = (
+        "legate --cpus=2 --gpus=0 --omps=0"
+        " legateboost_scaling.py --nrows 100 --ncols 5 --niter 2"
+        " --model_types tree,linear,krr,nn"
+    )
     res = subprocess.run(
-        cmd, shell=True, capture_output=True,
-        cwd=benchmark_dir, env=env
+        cmd, shell=True, capture_output=True, cwd=benchmark_dir, env=env
     )
-    assert res.returncode == 0, res.stderr.decode('utf-8')
+    assert res.returncode == 0, res.stderr.decode("utf-8")
 
 
 def assert_docstrings_run(files):
