Ensure Plane.mask ∈ {0, 1}

andykee · Feb 22, 2024 · da16374 · da16374
1 parent 0c4d70a
commit da16374
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lentil/plane.py b/lentil/plane.py
@@ -117,6 +117,7 @@ def mask(self):
     def mask(self, value):
         if value is not None:
             self._mask = np.asarray(value)
+            self._mask[self.mask != 0] = 1
         else:
             self._mask = None