wip

keithasaurus · Mar 1, 2024 · 97fd931 · 97fd931
1 parent 825fd8e
commit 97fd931
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/koda_validate/errors.py b/koda_validate/errors.py
@@ -161,7 +161,9 @@ class TypeErr:
     KeyErrs,
     MapErr,
     MissingKeyErr,
-    PredicateErrs,
+    # This seems like a type exception worth making..., but that might change in the
+    # future. This is backwards compatible with existing code
+    PredicateErrs[Any],
     SetErrs,
     TypeErr,
     ValidationErrBase,

diff --git a/koda_validate/signature.py b/koda_validate/signature.py
@@ -381,11 +381,13 @@ def validate_signature(
     if func is None:
 
         def inner(func_inner: _DecoratedFunc) -> _DecoratedFunc:
-            return _wrap_fn_partial(func_inner)
+            # there may be a good way to replace this cast with ParamSpec in the future
+            return cast(_DecoratedFunc, _wrap_fn_partial(func_inner))
 
         return inner
     else:
-        return _wrap_fn_partial(func)
+        # there may be a good way to replace this cast with ParamSpec in the future
+        return cast(_DecoratedFunc, _wrap_fn_partial(func))
 
 
 def _trunc_str(s: str, max_chars: int) -> str: