Fixes.

xarray-contrib · Nov 7, 2023 · ee62e37 · ee62e37
1 parent a5dc574
commit ee62e37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flox/core.py b/flox/core.py
@@ -1779,8 +1779,8 @@ def _choose_engine(by, agg: Aggregation):
 
     # numbagg only supports nan-skipping reductions
     # without dtype specified
-    has_blockwise_nan_skipping = (agg.chunk is None and "nan" in agg.name) or any(
-        "nan" in func for func in agg.chunk
+    has_blockwise_nan_skipping = (agg.chunk[0] is None and "nan" in agg.name) or any(
+        (isinstance(func, str) and "nan" in func) for func in agg.chunk
     )
     if HAS_NUMBAGG:
         if agg.name in ["all", "any"] or (