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

from . import builder_demo_base as base


class LinearPatch(pgml.YamlBaseModel):
    """Linear layer configuration"""

    in_features: int | None = None
    """Number of input features"""

    out_features: int | None = None
    """Number of output features"""

    bias: bool | None = None
    """Bias"""



class MLPPatch(pgml.YamlBaseModel):
    """MLP configuration"""

    layers: list[LinearPatch] | None = None
    """Linear layers"""


