Fix typing

tarsil · Oct 11, 2023 · dc5fb44 · dc5fb44
1 parent abccb2c
commit dc5fb44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/polyforce/_internal/_construction.py b/polyforce/_internal/_construction.py
@@ -51,13 +51,13 @@ def __new__(
             model = super().__new__(cls, name, bases, attrs)
             parents = [parent for parent in bases if isinstance(parent, PolyMetaclass)]
             if not parents:
-                return model
+                return cast("Type[PolyModel]", model)
 
             model = cast(Type["PolyModel"], model)
             complete_poly_class(model, config_wrapper)
             return model
 
-        return super().__new__(cls, name, bases, attrs)
+        return cast("Type[PolyModel]", super().__new__(cls, name, bases, attrs))
 
     @staticmethod
     def _collect_data_from_bases(bases: Tuple[Type]) -> Set[str]: