Modified ModelBase for track_overflow

hiyoneda · Nov 8, 2024 · d577f15 · d577f15
1 parent 54999aa
commit d577f15
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cosipy/image_deconvolution/model_base.py b/cosipy/image_deconvolution/model_base.py
@@ -16,10 +16,10 @@ class ModelBase(Histogram, ABC):
     """
 
     def __init__(self, edges, contents = None, sumw2 = None,
-                 labels=None, axis_scale = None, sparse = None, unit = None):
+                 labels=None, axis_scale = None, sparse = None, unit = None, track_overflow = False):
 
         super().__init__(edges, contents = contents, sumw2 = sumw2, 
-                         labels = labels, axis_scale = axis_scale, sparse = sparse, unit = unit)
+                         labels = labels, axis_scale = axis_scale, sparse = sparse, unit = unit, track_overflow = track_overflow)
 
     @classmethod
     @abstractmethod