Fix bug amplitude

IvoVellekoop · Oct 2, 2024 · 7e31402 · 7e31402
1 parent 1fb4e80
commit 7e31402
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."
             )