From 808610412316ab9f7125ce2331c90dfb8f4c9ce6 Mon Sep 17 00:00:00 2001 From: Ricky O'Steen Date: Fri, 24 May 2024 14:29:10 -0400 Subject: [PATCH] This seems inelegant but does fix the failing test --- specutils/analysis/moment.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/specutils/analysis/moment.py b/specutils/analysis/moment.py index 1dd4150e3..15971b58c 100644 --- a/specutils/analysis/moment.py +++ b/specutils/analysis/moment.py @@ -5,7 +5,7 @@ import numpy as np from ..manipulation import extract_region -from ..spectra import SpectralAxis +from ..spectra import SpectrumCollection from .utils import computation_wrapper @@ -40,6 +40,9 @@ def moment(spectrum, regions=None, order=0, axis=-1): Moment of the spectrum. Returns None if (order < 0 or None) """ + if isinstance(spectrum, SpectrumCollection): + return [computation_wrapper(_compute_moment, spec, regions,order=order, axis=axis) + for spec in spectrum] return computation_wrapper(_compute_moment, spectrum, regions, order=order, axis=axis)