From c5eb86af833989cea52900df29dc90e415b6a25d Mon Sep 17 00:00:00 2001 From: Eli Rykoff Date: Mon, 22 Jun 2026 14:29:32 -0700 Subject: [PATCH] Add additional info about max number of cosmic ray spans. --- python/lsst/pipe/tasks/repair.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/python/lsst/pipe/tasks/repair.py b/python/lsst/pipe/tasks/repair.py index caf6d6649..1dbb72b83 100644 --- a/python/lsst/pipe/tasks/repair.py +++ b/python/lsst/pipe/tasks/repair.py @@ -40,13 +40,16 @@ class TooManyCosmicRays(pipeBase.AlgorithmError): ---------- maxCosmicRays : `int` Maximum number of cosmic rays allowed. + maxCosmicRaySpans : `int` + Maximum number of cosmic ray spans allowed. """ - def __init__(self, maxCosmicRays, **kwargs): + def __init__(self, maxCosmicRays, maxCosmicRaySpans=-1, **kwargs): msg = f"Cosmic ray task found more than {maxCosmicRays} cosmics." self.msg = msg self._metadata = kwargs super().__init__(msg, **kwargs) self._metadata["maxCosmicRays"] = maxCosmicRays + self._metadata["maxCosmicRaySpans"] = maxCosmicRaySpans def __str__(self): # Exception doesn't handle **kwargs, so we need a custom str. @@ -231,7 +234,10 @@ def cosmicRay(self, exposure, keepCRs=None): keepCRs, ) except LengthError: - raise TooManyCosmicRays(self.config.cosmicray.nCrPixelMax) from None + raise TooManyCosmicRays( + self.config.cosmicray.nCrPixelMax, + maxCosmicRaySpans=self.config.cosmicray.nCrSpanMax, + ) from None if modelBg: # Add back background image