mirror of
https://github.com/wassname/jaxtyping.git
synced 2026-09-09 11:24:55 +08:00
Make test/helpers.py a relative import
This commit is contained in:
+2
-1
@@ -20,10 +20,11 @@
|
||||
import jax.numpy as jnp
|
||||
import jax.random as jr
|
||||
import pytest
|
||||
from helpers import ParamError, ReturnError
|
||||
|
||||
from jaxtyping import Array, f, f32, jaxtyped
|
||||
|
||||
from .helpers import ParamError, ReturnError
|
||||
|
||||
|
||||
def test_basic(typecheck):
|
||||
@jaxtyped
|
||||
|
||||
Reference in New Issue
Block a user