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

Make test/helpers.py a relative import

Browse Source
This commit is contained in:
Patrick Kidger
2022-07-12 10:03:19 +01:00
parent 81238e38e8
commit 9f75958b2d
8 changed files with 20 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_pytree.py
+2 -1
View File
@@ -24,10 +24,11 @@ import jax
import jax.numpy as jnp
import jax.random as jr
import pytest
from helpers import make_mlp, ParamError
from jaxtyping import f, jaxtyped, PyTree
from .helpers import make_mlp, ParamError
def test_direct(typecheck):
@typecheck