Hack to get sparse tests passing

pydata · Dec 31, 2020 · 83a3b89 · 83a3b89
1 parent 510a6e1
commit 83a3b89
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/xarray/core/dataset.py b/xarray/core/dataset.py
@@ -3832,7 +3832,12 @@ def unstack(
 
         result = self.copy(deep=False)
         for dim in dims:
-            if sparse:
+            # FIXME: remove
+            import sparse as sparse_
+
+            if sparse or any(
+                isinstance(v.data, sparse_.COO) for v in self.variables.values()
+            ):
                 result = result._unstack_once(dim, fill_value, sparse)
             else:
                 result = result._unstack_once_fast(dim, fill_value, sparse)

diff --git a/xarray/core/variable.py b/xarray/core/variable.py
@@ -1597,8 +1597,6 @@ def _unstack_once_fast(
         new_shape = list(reordered.shape[: len(other_dims)]) + new_dim_sizes
         new_dims = reordered.dims[: len(other_dims)] + new_dim_names
 
-        # missing_values = np.prod(new_shape) > np.prod(self.shape)
-
         if fill_value is dtypes.NA:
             fill_value = dtypes.get_fill_value(self.dtype)