DelayedMask returns True for is_masked, duh.

ArtifactDB · Jan 29, 2024 · f1ba4f6 · f1ba4f6
1 parent a96a4c5
commit f1ba4f6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/dolomite_matrix/DelayedMask.py b/src/dolomite_matrix/DelayedMask.py
@@ -134,3 +134,8 @@ def is_sparse_DelayedMask(x: DelayedMask):
     """See :py:meth:`~delayedarray.is_sparse.is_sparse`."""
     return delayedarray.is_sparse(x._seed)
 
+
+@delayedarray.is_masked.register
+def is_masked_DelayedMask(x: DelayedMask):
+    """See :py:meth:`~delayedarray.is_masked.is_masked`."""
+    return True
diff --git a/tests/test_DelayedMask.py b/tests/test_DelayedMask.py
@@ -11,6 +11,7 @@ def test_DelayedMask_dense():
     assert m.shape == y.shape
     assert m.placeholder == 1
     assert not delayedarray.is_sparse(m)
+    assert delayedarray.is_masked(m)
     assert delayedarray.chunk_shape(m) == (1, 3)
 
     block = delayedarray.extract_dense_array(m)