Merge pull request #547 from lsst/tickets/DM-30959

DM-30959: Prevent maskStreaks failure due to unmasked nans or infs

python/lsst/pipe/tasks/maskStreaks.py

@@ -405,6 +405,10 @@ def line_search(c, dx):
             chi2, b, A = self._lineChi2(line)
             if chi2 == 0:
                 break
+            if not np.isfinite(A).all():
+                # TODO: DM-30797 Add warning here.
+                fitFailure = True
+                break
             dChi2 = oldChi2 - chi2
             cholesky = scipy.linalg.cho_factor(A)
             dx = scipy.linalg.cho_solve(cholesky, b)
@@ -708,6 +712,8 @@ def _fitProfile(self, lines, maskedImage):
         """
         data = maskedImage.image.array
         weights = maskedImage.variance.array**-1
+        # Mask out any pixels with non-finite weights
+        weights[~np.isfinite(weights) | ~np.isfinite(data)] = 0
 
         lineFits = LineCollection([], [])
         finalLineMasks = [np.zeros(data.shape, dtype=bool)]