diff --git a/tests/ert/ui_tests/cli/conftest.py b/tests/ert/ui_tests/cli/conftest.py index e04d2bd3764..e99b3121999 100644 --- a/tests/ert/ui_tests/cli/conftest.py +++ b/tests/ert/ui_tests/cli/conftest.py @@ -6,7 +6,9 @@ @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 @@ -14,3 +16,7 @@ def reduce_omp_num_threads_count(): 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