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

Added jaxtyping.Array=jnp.ndarray

Browse Source
This commit is contained in:
Patrick Kidger
2022-08-30 12:56:47 -07:00
parent 903000f3d5
commit 48e8131247
4 changed files with 27 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_array.py
+1 -4
View File
@@ -21,14 +21,11 @@ import jax.numpy as jnp
import jax.random as jr
import pytest
from jaxtyping import f32, Float, jaxtyped, Shaped
from jaxtyping import Array, f32, Float, jaxtyped, Shaped
from .helpers import ParamError, ReturnError
Array = jnp.ndarray
def test_basic(typecheck):
@jaxtyped
@typecheck