Skip to content

Commit 87bb0ec

Browse files
committed
feat(audio-advanced_pipelines): make advanced_pipelines stages agent-ready (describe/StageContract + residency)
1 parent d019ecf commit 87bb0ec

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

nemo_curator/stages/audio/advanced_pipelines/audio_data_filter/audio_data_filter.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646

4747
from loguru import logger
4848

49+
from nemo_curator.stages.audio._agent_ready import AgentReady, StageContract
4950
from nemo_curator.stages.audio.filtering import BandFilterStage, SIGMOSFilterStage, UTMOSFilterStage
5051
from nemo_curator.stages.audio.postprocessing import TimestampMapperStage
5152
from nemo_curator.stages.audio.preprocessing import MonoConversionStage, SegmentConcatenationStage
@@ -56,7 +57,7 @@
5657
from .config import _deep_merge, get_enabled_stages, load_config
5758

5859

59-
class AudioDataFilterStage(CompositeStage):
60+
class AudioDataFilterStage(AgentReady, CompositeStage):
6061
"""Complete audio data filtering and curation pipeline (CompositeStage).
6162
6263
Decomposes into independent stages that the executor can schedule with
@@ -90,6 +91,9 @@ def __init__(
9091
if config:
9192
self._cfg = _deep_merge(self._cfg, config)
9293

94+
def describe(self) -> StageContract:
95+
return StageContract(wrappable=False)
96+
9397
def decompose(self) -> list[ProcessingStage]:
9498
"""Build a self-consistent pipeline topology based on enabled features."""
9599
cfg = self._cfg

0 commit comments

Comments
 (0)