disable dask query-planning as workaround for bugs in dask-expr

OpenCOMPES · Oct 11, 2024 · 1aa4a83 · 1aa4a83
1 parent daf5c2d
commit 1aa4a83
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sed/__init__.py b/sed/__init__.py
@@ -1,7 +1,11 @@
 """sed module easy access APIs."""
 import importlib.metadata
 
-from .core.processor import SedProcessor
+import dask
+
+dask.config.set({"dataframe.query-planning": False})
+
+from .core.processor import SedProcessor  # noqa: E402
 
 __version__ = importlib.metadata.version("sed-processor")
 __all__ = ["SedProcessor"]