From 270864b7476535ccdadb400fabdf5df05b2973af Mon Sep 17 00:00:00 2001
From: nstarman <nstarman@users.noreply.github.com>
Date: Sat, 9 Mar 2024 14:55:41 -0500
Subject: [PATCH] fix: import

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
---
 tests/myarray.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/myarray.py b/tests/myarray.py
index fef6031..54fe273 100644
--- 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):