Fixed jaxtyped breaking descriptors. Fixed long module names. (#25)

2026-09-09 11:24:55 +08:00 · 2022-09-19 22:40:25 -07:00
parent 62ddcc25b5
commit c2e9d913d5
5 changed files with 35 additions and 8 deletions
@@ -0,0 +1,13 @@
+from jaxtyping import jaxtyped
+
+
+class M:
+    @jaxtyped
+    @classmethod
+    def f(cls):
+        return 3
+
+
+# Check that the @jaxtyped decorator doesn't blat the __get__ of @classmethod
+def test_decorator():
+    assert M.f() == 3