Now more robust to errors during type hint resolution.

Author: patrick-kidger

jaxtyping/_decorator.py

@@ -393,8 +393,12 @@ def wrapped_fn(*args, **kwargs):  # pyright: ignore
             full_signature = inspect.signature(fn)
             try:
                 destring_annotations = get_type_hints(fn, include_extras=True)
-            except NameError:
+            except Exception:
                 # Best-effort attempt to destringify annotations.
+                # Not just `NameError` but also e.g. `ValueError` in case we have e.g.
+                # 'Float[Foo, "*foo *bar"]' and raise  from having multiple variadic
+                # arguments. Sometimes this can still be useful to use for human
+                # documentation purposes.
                 pass
             else:
                 new_params = []

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "jaxtyping"
-version = "0.3.1"
+version = "0.3.2"
 description = "Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays."
 readme = "README.md"
 requires-python =">=3.10"

test/test_decorator.py

@@ -305,3 +305,11 @@ def __init__(self, x: int):
     wrapped = Foo.__init__.__wrapped__
     with pytest.raises(AttributeError):
         wrapped.__wrapped__
+
+
+def test_stringified_multiple_varaidic(typecheck):
+    @jaxtyped(typechecker=typecheck)
+    def foo() -> 'Float[Array, "*foo *bar"]':
+        return jnp.arange(3)
+
+    foo()