stages.audio.io.convert

Module Contents

Classes

AudioToDocumentStage

Stage to conver DocumentObject to DocumentBatch

API

class stages.audio.io.convert.AudioToDocumentStage

Bases: nemo_curator.stages.base.ProcessingStage[nemo_curator.tasks.AudioBatch, nemo_curator.tasks.DocumentBatch]

Stage to conver DocumentObject to DocumentBatch

process(

task: nemo_curator.tasks.AudioBatch,

) → list[nemo_curator.tasks.DocumentBatch]

Process a task and return the result. Args: task (X): Input task to process Returns (Y | list[Y]): - Single task: For 1-to-1 transformations - List of tasks: For 1-to-many transformations (e.g., readers) - None: If the task should be filtered out