data_designer.engine.processing.processors.base
data_designer.engine.processing.processors.base
data_designer.engine.processing.processors.base
Bases: data_designer.engine.configurable_task.ConfigurableTask[data_designer.engine.configurable_task.TaskConfigT], abc.ABC
Base class for dataset processors.
Processors transform data at different stages of the generation pipeline. Override the callback methods for the stages you want to handle.
Check if subclass overrides a callback method.
Called at PRE_BATCH stage before each batch is generated.
Override to transform batch data before generation begins.
Parameters:
The batch data before generation.
Returns:
data_designer.engine.configurable_task.DataT
Transformed batch data.
Called at POST_BATCH stage after each batch is generated.
Override to process each batch of generated data.
Parameters:
The generated batch data.
The current batch number (0-indexed), or None in preview mode.
Returns:
data_designer.engine.configurable_task.DataT
Transformed batch data.
Called at AFTER_GENERATION stage on the final combined dataset.
Override to transform the complete generated dataset.
Parameters:
The final combined dataset.
Returns:
data_designer.engine.configurable_task.DataT
Transformed final dataset.