Fix bug amplitude

IvoVellekoop · Oct 2, 2024 · b9a944b · b9a944b
1 parent 1fb4e80
commit b9a944b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openwfs/algorithms/dual_reference.py b/openwfs/algorithms/dual_reference.py
@@ -126,7 +126,7 @@ def amplitude(self, value):
                 (np.ones(shape=self._shape) / np.sqrt(self.masks[side].sum())).astype(np.float32) for side in range(2))
             return
 
-        if value.shape != self._shape:
+        if value[0].shape != self._shape or value[1].shape != self._shape :
             raise ValueError(
                 "The amplitude and group mask must all have the same shape."
             )