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

import pydantic
import typer

from . import default_base as base


class OptimizerArgs(pgml.YamlBaseModel):
    """Optimizer configuration"""

    type: typing.Annotated[typing.Optional[str], typer.Option(help="Optimizer type. Maps to 'type'"), pydantic.Field(None), pgml.ArgRef("type")]
    """Optimizer type"""

    learning_rate: typing.Annotated[typing.Optional[float], typer.Option(help="Learning rate. Maps to 'learning_rate'"), pydantic.Field(None), pgml.ArgRef("learning_rate")]
    """Learning rate"""


