fix: import

Signed-off-by: nstarman <[email protected]>
GalacticDynamics · Mar 9, 2024 · 270864b · 270864b
1 parent 15de0a6
commit 270864b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/myarray.py b/tests/myarray.py
@@ -14,8 +14,8 @@
 from jaxtyping import ArrayLike
 from quax import ArrayValue, register
 
-from quaxed._dispatch import dispatcher
-from quaxed._types import DType
+from quaxed.array_api._dispatch import dispatcher
+from quaxed.array_api._types import DType
 
 
 class MyArray(ArrayValue):