Symbolic expressions now support delayed binding to arguments. Fixes #93.

2026-09-13 12:40:59 +08:00 · 2023-11-27 09:50:02 -08:00
parent ba3b2027cc
commit 7925e278f4
7 changed files with 160 additions and 56 deletions
@@ -88,6 +88,8 @@ def test_context(getkey):
    with jaxtyped("context"):
        assert isinstance(a, Float[Array, "foo bar"])
        assert not isinstance(b, Float[Array, "foo"])
+    assert isinstance(a, Float[Array, "foo bar"])
+    assert isinstance(b, Float[Array, "foo"])


 def test_varargs(jaxtyp, typecheck):