# Autogenerated code. DO NOT EDIT.
import py_gen_ml as pgml

from . import cli_demo_base as base


class ModelPatch(pgml.YamlBaseModel):
    """Model configuration"""

    num_layers: int | None = None
    """Number of layers"""



class TrainingPatch(pgml.YamlBaseModel):
    """Training configuration"""

    num_epochs: int | None = None
    """Number of epochs"""



class DatasetPatch(pgml.YamlBaseModel):
    """Dataset configuration"""

    path: str | None = None
    """Path to the dataset"""



class DataPatch(pgml.YamlBaseModel):
    """Data config"""

    dataset: DatasetPatch | None = None
    """Path to the dataset"""

    num_workers: int | None = None
    """Number of workers for loading the dataset"""



class ProjectPatch(pgml.YamlBaseModel):
    """Global configuration"""

    data: DataPatch | None = None
    """Dataset configuration"""

    model: ModelPatch | None = None
    """Model configuration"""

    training: TrainingPatch | None = None
    """Training configuration"""


