Make test/helpers.py a relative import

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
+2 -1
View File
@@ -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