Update Wrappers/Python/cil/optimisation/functions/TotalVariation.py

Co-authored-by: Edoardo Pasca <edo.paskino@gmail.com> Signed-off-by: Margaret Duff <43645617+MargaretDuff@users.noreply.github.com>
TomographicImaging · Aug 8, 2023 · 647dd41 · 647dd41
1 parent d39387b
commit 647dd41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Wrappers/Python/cil/optimisation/functions/TotalVariation.py b/Wrappers/Python/cil/optimisation/functions/TotalVariation.py
@@ -225,7 +225,7 @@ def p2(self):
 
     @p2.setter
     def p2(self, value):
-        if isinstance(value, type(self.gradient.range_geometry().allocate(0))):
+        if len(value.geometry.geometries) != len(self.gradient.range_geometry().geometries):
             self._p2 = value
         else:
             raise TypeError('p2 should be in the range of the gradient of the image')