add ignores for mypy

Author: svlandeg

sqlmodel/main.py

@@ -685,7 +685,7 @@ def __init__(
             table = getattr(cls, "__table__", None)
             if table is not None:
                 # Attach Literal-based value constraints at the database level
-                for field_name, field in get_model_fields(cls).items():
+                for field_name, field in get_model_fields(cls).items():  # type: ignore
                     annotation = getattr(field, "annotation", None)
                     literal_info = get_literal_annotation_info(annotation)
                     if literal_info is None:
@@ -700,7 +700,7 @@ def __init__(
                     elif base_type is bool:
                         coerced_values = tuple(bool(v) for v in values)
                     else:
-                        coerced_values = tuple(str(v) for v in values)
+                        coerced_values = tuple(str(v) for v in values)  # type: ignore
                     constraint_name = f"ck_{table.name}_{field_name}_literal"
                     constraint = CheckConstraint(
                         column.in_(coerced_values),

tests/test_main.py

@@ -1,4 +1,4 @@
-from typing import Annotated, Literal, Union
+from typing import Annotated, Literal
 
 import pytest
 from sqlalchemy import text
@@ -332,7 +332,7 @@ def test_literal_optional_and_union_constraints(clear_sqlmodel):
     class Model(SQLModel, table=True):
         id: int | None = Field(default=None, primary_key=True)
         opt_str: Literal["x", "y"] | None = None
-        union_int: Union[Literal[10, 20], None] = None
+        union_int: Literal[10, 20] | None = None
 
     engine = create_engine("sqlite://")
     SQLModel.metadata.create_all(engine)