wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-09-09 11:24:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Tidied up _#* into acting as prefixes, regardless of order. Closes #10

Browse Source
This commit is contained in:
Patrick Kidger
2022-07-31 23:45:40 +01:00
parent 7b698ae215
commit 6e2837e5b7
5 changed files with 111 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_array.py
+3 -3
View File
@@ -227,7 +227,7 @@ def test_anonymous_variadic(typecheck, getkey):
def test_broadcast_fixed(typecheck, getkey):
@jaxtyped
@typecheck
def g(x: f32["4#"]):
def g(x: f32["#4"]):
pass
g(jr.normal(getkey(), (4,)))
@@ -240,7 +240,7 @@ def test_broadcast_fixed(typecheck, getkey):
def test_broadcast_named(typecheck, getkey):
@jaxtyped
@typecheck
def g(x: f32[" foo#"], y: f32[" foo#"]):
def g(x: f32[" #foo"], y: f32[" #foo"]):
pass
a = jr.normal(getkey(), (3,))
@@ -264,7 +264,7 @@ def test_broadcast_named(typecheck, getkey):
def test_broadcast_variadic_named(typecheck, getkey):
@jaxtyped
@typecheck
def g(x: f32[" *foo#"], y: f32[" *foo#"]):
def g(x: f32[" *#foo"], y: f32[" *#foo"]):
pass
a = jr.normal(getkey(), (3,))