From 234b8d75290efb7d2d4505e567e63aaf4a4accd8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Simon=20H=C3=B8xbro=20Hansen?= <simon.hansen@me.com>
Date: Thu, 4 Apr 2024 13:10:21 +0200
Subject: [PATCH] Set environment variables for query-planning

---
 .github/workflows/test.yaml | 15 +++++++++------
 examples/conftest.py        | 10 ----------
 2 files changed, 9 insertions(+), 16 deletions(-)

diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index 09fdaa5fb..cfe40fd1a 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -27,6 +27,15 @@ concurrency:
   group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}
   cancel-in-progress: true
 
+env:
+  OMP_NUM_THREADS: 1
+  OPENBLAS_NUM_THREADS: 1
+  MKL_NUM_THREADS: 1
+  VECLIB_MAXIMUM_THREADS: 1
+  NUMEXPR_NUM_THREADS: 1
+  PYDEVD_DISABLE_FILE_VALIDATION: 1
+  DASK_DATAFRAME__QUERY_PLANNING: false
+
 jobs:
   pre_commit:
     name: Run pre-commit
@@ -90,12 +99,6 @@ jobs:
         shell: bash -l {0}
     env:
       GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-      OMP_NUM_THREADS: 1
-      OPENBLAS_NUM_THREADS: 1
-      MKL_NUM_THREADS: 1
-      VECLIB_MAXIMUM_THREADS: 1
-      NUMEXPR_NUM_THREADS: 1
-      PYDEVD_DISABLE_FILE_VALIDATION: 1
     steps:
       - uses: holoviz-dev/holoviz_tasks/install@v0
         with:
diff --git a/examples/conftest.py b/examples/conftest.py
index 6e5591fd6..1eb79c649 100644
--- a/examples/conftest.py
+++ b/examples/conftest.py
@@ -1,6 +1,4 @@
 import platform
-import contextlib
-
 
 collect_ignore_glob = []
 
@@ -21,11 +19,3 @@
     collect_ignore_glob += [
         "user_guide/7_Networks.ipynb",
     ]
-
-
-with contextlib.suppress(Exception):
-    # From Dask 2024.3.0 they now use `dask_expr` by default
-    # https://github.com/dask/dask/issues/10995
-    import dask
-
-    dask.config.set({"dataframe.query-planning": False})