Don't trigger dask_graph call all the time

mrocklin · Dec 8, 2023 · 15c7c0f · 15c7c0f
1 parent d30b697
commit 15c7c0f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dask/base.py b/dask/base.py
@@ -211,7 +211,11 @@ def is_dask_collection(x) -> bool:
 
     """
     try:
-        return x.__dask_graph__() is not None
+        if "xarray" in type(x).__module__:
+            return x.__dask_graph__() is not None
+        else:
+            return hasattr(x, "__dask_graph__")
+
     except (AttributeError, TypeError):
         return False