diff --git a/mypy_django_plugin/helpers.py b/mypy_django_plugin/helpers.py index ee3a98b..90e49b4 100644 --- a/mypy_django_plugin/helpers.py +++ b/mypy_django_plugin/helpers.py @@ -254,7 +254,9 @@ def extract_primary_key_type_for_get(model: TypeInfo) -> Optional[Type]: return None -def make_optional(typ: Type): +def make_optional(typ: Type) -> UnionType: + if isinstance(typ, AnyType): + return typ return UnionType.make_union([typ, NoneTyp()])