Just use utility function _dask_expr_enabled in liniar_model/utils.py

dask · Mar 5, 2024 · 98d150c · 98d150c
1 parent 864e5ad
commit 98d150c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/dask_ml/linear_model/utils.py b/dask_ml/linear_model/utils.py
@@ -1,6 +1,5 @@
 """
 """
-import dask
 import dask.array as da
 import dask.dataframe as dd
 import numpy as np
@@ -11,7 +10,7 @@
 except ImportError:
     dask_expr = None
 
-if dask.config.get("dataframe.query-planning", True) and dask_expr is not None:
+if getattr(dask_expr, "_dask_expr_enabled", lambda: False)():
 
     @dispatch(dask_expr.FrameBase)
     def exp(A):