Testing in the p2 setter

TomographicImaging · Aug 8, 2023 · bc7ae27 · bc7ae27
1 parent 7d539fe
commit bc7ae27
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
@@ -228,7 +228,7 @@ def p2(self):
     @p2.setter
     def p2(self, value):
         r""" Setter for the dual value in the proximal calculation."""
-        if len(value.geometry.geometries) != len(self.gradient.range_geometry().geometries):
+        if len(value) != len(self.gradient.range_geometry().geometries):
             self._p2 = value
         else:
             raise TypeError('p2 should be in the range of the gradient of the image')