Refactor model signature creation and transform

tarsil · Oct 11, 2023 · 48aa41b · 48aa41b
1 parent 98e536e
commit 48aa41b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/polyforce/_internal/_construction.py b/polyforce/_internal/_construction.py
@@ -71,8 +71,6 @@ def generate_model_signature(
             param = param.replace(annotation=Any)
         elif param.annotation in config.ignored_types:
             param = param.replace(annotation=Any)
-        if param.kind is param.VAR_KEYWORD:
-            continue
         merged_params[param.name] = param
 
     # Generate the new signatures.

diff --git a/polyforce/main.py b/polyforce/main.py
@@ -77,7 +77,7 @@ def __getattribute__(self, name: str) -> Any:
         except (KeyError, AttributeError):
             return object.__getattribute__(self, name)
 
-    def _extract_type_hint(type_hint: Union[Type, tuple]) -> Union[Type, tuple]:
+    def _extract_type_hint(self, type_hint: Union[Type, tuple]) -> Union[Type, tuple]:
         """
         Extracts the base type from a type hint, considering typing extensions.