Set polars max threads to 1 in cli tests

larsevj · Nov 27, 2024 · 659af57 · 659af57
1 parent 0b34fd5
commit 659af57
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/ert/ui_tests/cli/conftest.py b/tests/ert/ui_tests/cli/conftest.py
@@ -6,11 +6,17 @@
 @pytest.fixture(autouse=True)
 def reduce_omp_num_threads_count():
     old_omp_num_threads = os.environ.get("OMP_NUM_THREADS")
+    old_polars_max_thread = os.environ.get("POLARS_MAX_THREADS")
     os.environ["OMP_NUM_THREADS"] = "1"
+    os.environ["POLARS_MAX_THREADS"] = "1"
 
     yield
 
     if old_omp_num_threads is None:
         del os.environ["OMP_NUM_THREADS"]
     else:
         os.environ["OMP_NUM_THREADS"] = old_omp_num_threads
+    if old_polars_max_thread is None:
+        del os.environ["POLARS_MAX_THREADS"]
+    else:
+        os.environ["POLARS_MAX_THREADS"] = old_polars_max_thread