diff options
-rwxr-xr-x | Wrappers/Python/ccpi/optimisation/algorithms/CGLS.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/Wrappers/Python/ccpi/optimisation/algorithms/CGLS.py b/Wrappers/Python/ccpi/optimisation/algorithms/CGLS.py index 4faffad..15acc31 100755 --- a/Wrappers/Python/ccpi/optimisation/algorithms/CGLS.py +++ b/Wrappers/Python/ccpi/optimisation/algorithms/CGLS.py @@ -106,9 +106,7 @@ class CGLS(Algorithm): self.d *= alpha Ad *= alpha self.r -= Ad - if numpy.isnan(self.r.as_array()).any(): - print ("some nan") - raise StopIteration() + self.x += self.d self.operator.adjoint(self.r, out=self.s) |