INNER CODE UNIT · Python

GenerationPreparation

nolabs-ai/deepfabric · deepfabric/cli.py:176

class GenerationPreparation(BaseModel):
    """Validated state required to run dataset generation."""

    model_config = ConfigDict(arbitrary_types_allowed=True)

    config: DeepFabricConfig
    topics_overrides: OverrideMap = Field(default_factory=dict)
    generation_overrides: OverrideMap = Field(default_factory=dict)
    num_samples: int | str  # Can be int, "auto", or percentage like "50%"
    batch_size: int
    depth: int
    degree: int
    loading_existing: bool

    @model_validator(mode="after")
    def validate_positive_dimensions(self) -> "GenerationPreparation":
        # Skip num_samples validation for dynamic values (auto or percentage)
        if isinstance(self.num_samples, int) and self.num_samples <= 0:

View source record →