diff --git a/src/ezmsg/sigproc/scaler.py b/src/ezmsg/sigproc/scaler.py index 79b0648f..f1877acc 100644 --- a/src/ezmsg/sigproc/scaler.py +++ b/src/ezmsg/sigproc/scaler.py @@ -162,6 +162,8 @@ class AdaptiveStandardScaler( ): SETTINGS = AdaptiveStandardScalerSettings + INPUT_ACCUMULATE = ez.InputStream(bool) + @ez.subscriber(BaseTransformerUnit.INPUT_SETTINGS) async def on_settings(self, msg: AdaptiveStandardScalerSettings) -> None: """ @@ -183,6 +185,10 @@ async def on_settings(self, msg: AdaptiveStandardScalerSettings) -> None: # Also update own settings reference self.processor.settings = msg + @ez.subscriber(INPUT_ACCUMULATE) + async def on_accumulate(self, accumulate: bool) -> None: + self.processor.accumulate = accumulate + # Convenience functions to support deprecated generator API def scaler(time_constant: float = 1.0, axis: str | None = None) -> RiverAdaptiveStandardScalerTransformer: