From 7e31402948d9079870950947cef4be15baff5257 Mon Sep 17 00:00:00 2001 From: Daniel Cox Date: Wed, 2 Oct 2024 17:48:36 +0200 Subject: [PATCH] Fix bug amplitude --- openwfs/algorithms/dual_reference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openwfs/algorithms/dual_reference.py b/openwfs/algorithms/dual_reference.py index a16d2f7..8ac2703 100644 --- 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." )