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