optimize

Author: dcherian

xarray/core/dataset.py

@@ -293,10 +293,10 @@ def _get_breaks_cached(
     # Determine the stop indices of the preferred chunks, but omit the last stop
     # (equal to the dim size).  In particular, assume that when a sequence
     # expresses the preferred chunks, the sequence sums to the size.
-    preferred_stops = set(
+    preferred_stops = (
         range(preferred_chunk_sizes, size, preferred_chunk_sizes)
         if isinstance(preferred_chunk_sizes, int)
-        else itertools.accumulate(preferred_chunk_sizes[:-1])
+        else set(itertools.accumulate(preferred_chunk_sizes[:-1]))
     )
 
     # Gather any stop indices of the specified chunks that are not a stop index