schemas.py.jinja

# Automatically generated file. Do not modify.

import pandera as pa
from pandera.dtypes import Int32, Timestamp
from pandera.typing import Series


class _BaseSchema(pa.DataFrameModel):
    class Config:
        add_missing_columns = True
        coerce = True


{% for m in models %}
class {{m[:name]}}Schema(_BaseSchema):
  {% for f in m[:fields] %}
    {% if (f[2] == "Series[Int32]") %}
    {{ f[1] }}: {{ f[2] }} = pa.Field(nullable={{ f[3] }}, default=0)
    {% else %}
    {{ f[1] }}: {{ f[2] }} = pa.Field(nullable={{ f[3] }})
    {% end %}
  {% end %}
{% end %}