Fix str->int type annotation

patrick-kidger · patrick-kidger · commit 682787f071ee · 2025-02-24T10:41:46.000+01:00
diff --git a/jaxtyping/_array_types.py b/jaxtyping/_array_types.py
@@ -98,7 +98,7 @@ class _NamedVariadicDim:
 
 @dataclass(frozen=True)
 class _FixedDim:
-    size: str
+    size: int
     broadcastable: bool
 
 
