refactor: use only a valid_mask for simplifying (#91)

UNDP-Data · Apr 9, 2024 · 6931206 · 6931206
1 parent f8223c7
commit 6931206
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/cogserver/algorithms/rca.py b/src/cogserver/algorithms/rca.py
@@ -79,9 +79,12 @@ def __call__(self, img: ImageData) -> ImageData:
         v = 0.1
         #v = data.ptp()*.1
         datam = (data > -v) & (data < v)
+        valid_mask |= datam
         if self.threshold:
-            datam = datam | np.abs(data)< self.threshold
-        arr = numpy.ma.masked_array(data*100, dtype=self.output_dtype, mask=valid_mask | datam )
+            threshold_mask = np.abs(data)< self.threshold
+            valid_mask |= threshold_mask
+
+        arr = numpy.ma.masked_array(data*100, dtype=self.output_dtype, mask=valid_mask)
         return ImageData(
             arr,
             assets=img.assets,