Skip to content

Commit 346c59a

Browse files
committed
MAINT: fix up after the rebase
1 parent 0832120 commit 346c59a

3 files changed

Lines changed: 7 additions & 3 deletions

File tree

array_api_strict/_array_object.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,10 @@
4040
_real_to_complex_map,
4141
_result_type,
4242
)
43-
from ._devices import CPU_DEVICE, Device, device_supports_dtype
43+
from ._devices import (
44+
CPU_DEVICE, Device, device_supports_dtype, _normalize_dl_device, _DLPACK_DEVICE_FOR,
45+
DLDeviceType
46+
)
4447
from ._flags import get_array_api_strict_flags, set_array_api_strict_flags
4548
from ._typing import PyCapsule
4649

array_api_strict/_creation_functions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def from_dlpack(
247247
else:
248248
device = None
249249
if hasattr(x, "__dlpack_device__"):
250-
from ._array_object import _device_from_dlpack_device
250+
from ._devices import _device_from_dlpack_device
251251

252252
dl_type, dl_id = x.__dlpack_device__()
253253
device = _device_from_dlpack_device(dl_type, dl_id)

array_api_strict/tests/test_array_object.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@
99
import pytest
1010

1111
from .. import ones, arange, reshape, asarray, result_type, all, equal, stack
12-
from .._array_object import Array, CPU_DEVICE, Device, DLDeviceType
12+
from .._array_object import Array
13+
from .._devices import CPU_DEVICE, Device, DLDeviceType
1314
from .._dtypes import (
1415
_all_dtypes,
1516
_boolean_dtypes,

0 commit comments

Comments
 (0)