pydantic_models/MyModel.py

"""Template for custom function or Pydantic model."""

from pydantic import BaseModel
from pydantic import Field


class MyModel(BaseModel):
    """My model."""

    attr1: str = Field(..., description="The first attribute.")
    attr2: int = Field(..., description="The second attribute.")