maybe fix again

apache · Nov 10, 2023 · 7d2db62 · 7d2db62
1 parent 64185cb
commit 7d2db62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tests/test_nanoarrow_device.py b/python/tests/test_nanoarrow_device.py
@@ -18,7 +18,7 @@
 import pyarrow as pa
 
 import nanoarrow as na
-import nanoarrow.device as na_device
+from nanoarrow import device
 
 
 def test_cpu_device():
@@ -30,7 +30,7 @@ def test_cpu_device():
 
     pa_array = pa.array([1, 2, 3])
 
-    darray = na_device.device_array(pa_array)
+    darray = device.device_array(pa_array)
     assert darray.device_type == 1
     assert darray.device_id == 0
     assert darray.array.length == 3