fixup! Include ModelBase subclasses in plugin base class hook condition

typeddjango · Dec 8, 2023 · 983a752 · 983a752
1 parent 05116bf
commit 983a752
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/mypy_django_plugin/lib/helpers.py b/mypy_django_plugin/lib/helpers.py
@@ -474,9 +474,4 @@ def resolve_lazy_reference(
 
 
 def is_model_type(info: TypeInfo) -> bool:
-    return info.metaclass_type is not None and (
-        # Using the model metaclass shipped by Django
-        info.metaclass_type.type.fullname == fullnames.MODEL_METACLASS_FULLNAME
-        # Using a custom model metaclass that inherits above
-        or info.metaclass_type.type.has_base(fullnames.MODEL_METACLASS_FULLNAME)
-    )
+    return info.metaclass_type is not None and info.metaclass_type.type.has_base(fullnames.MODEL_METACLASS_FULLNAME)