fix typings

dymmond · Jan 12, 2025 · d6d6b87 · d6d6b87
1 parent fbb8dc6
commit d6d6b87
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/edgy/contrib/autoreflection/models.py b/edgy/contrib/autoreflection/models.py
@@ -12,7 +12,7 @@ class AutoReflectModel(edgy.ReflectModel, metaclass=AutoReflectionMeta):
     meta: ClassVar[AutoReflectionMetaInfo]
 
     @classmethod
-    def real_add_to_registry(cls: type["BaseModelType"], **kwargs: Any) -> type["BaseModelType"]:
+    def real_add_to_registry(cls, **kwargs: Any) -> type["BaseModelType"]:
         if isinstance(cls.meta, AutoReflectionMetaInfo):
             kwargs.setdefault("registry_type_name", "pattern_models")
         return super().real_add_to_registry(**kwargs)
diff --git a/edgy/contrib/multi_tenancy/base.py b/edgy/contrib/multi_tenancy/base.py
@@ -21,7 +21,7 @@ class TenantModel(Model, metaclass=BaseTenantMeta):
     meta: ClassVar[TenantMeta] = TenantMeta(None, abstract=True)
 
     @classmethod
-    def real_add_to_registry(cls: type["BaseModelType"], **kwargs: Any) -> type["BaseModelType"]:
+    def real_add_to_registry(cls, **kwargs: Any) -> type["BaseModelType"]:
         result = super().real_add_to_registry(**kwargs)
 
         if (
@@ -30,9 +30,9 @@ def real_add_to_registry(cls: type["BaseModelType"], **kwargs: Any) -> type["Bas
             and not cls.meta.abstract
             and not cls.__is_proxy_model__
         ):
-            assert (
-                cls.__reflected__ is False
-            ), "Reflected models are not compatible with multi_tenancy"
+            assert cls.__reflected__ is False, (
+                "Reflected models are not compatible with multi_tenancy"
+            )
 
             if not cls.meta.register_default:
                 # remove from models

diff --git a/edgy/core/db/models/mixins/db.py b/edgy/core/db/models/mixins/db.py
@@ -248,7 +248,7 @@ def create_through_model(x: Any, field: BaseFieldType = value) -> None:
 
     @classmethod
     def add_to_registry(
-        cls: type[BaseModelType],
+        cls,
         registry: Registry,
         name: str = "",
         database: Union[bool, Database, Literal["keep"]] = "keep",