diff --git a/python/lsst/meas/algorithms/accumulator_mean_stack.py b/python/lsst/meas/algorithms/accumulator_mean_stack.py index b978b15b1..d8676d34a 100644 --- a/python/lsst/meas/algorithms/accumulator_mean_stack.py +++ b/python/lsst/meas/algorithms/accumulator_mean_stack.py @@ -27,7 +27,7 @@ __all__ = ['AccumulatorMeanStack'] -class AccumulatorMeanStack(object): +class AccumulatorMeanStack: """Stack masked images. Parameters @@ -99,9 +99,8 @@ def add_masked_image(self, masked_image, weight=1.0): ---------- masked_image : `lsst.afw.image.MaskedImage` Masked image to add to the stack. - weight : `float` or `np.ndarray`, optional - Weight to apply for weighted mean. If an array, - must be same size and shape as input masked_image. + weight : `float`, optional + Weight to apply for weighted mean. """ good_pixels = np.where(((masked_image.mask.array & self.bit_mask_value) == 0) & np.isfinite(masked_image.mask.array)) @@ -187,9 +186,8 @@ def add_image(self, image, weight=1.0): ---------- image : `lsst.afw.image.Image` Image to add to the stack. - weight : `float` or `np.ndarray`, optional - Weight to apply for weighted mean. If an array, - must be same size and shape as input image. + weight : `float`, optional + Weight to apply for weighted mean. """ self.sum_weight[:, :] += weight self.sum_wdata[:, :] += weight*image.array[:]