diff --git a/array_api_compat/cupy/__init__.py b/array_api_compat/cupy/__init__.py index 558a83e1..246ac872 100644 --- a/array_api_compat/cupy/__init__.py +++ b/array_api_compat/cupy/__init__.py @@ -9,8 +9,8 @@ from ._info import __array_namespace_info__ # noqa: F401 # See the comment in the numpy __init__.py -__import__(__package__ + '.linalg') -__import__(__package__ + '.fft') +__import__(__spec__.parent + '.linalg') +__import__(__spec__.parent + '.fft') __array_api_version__: Final = '2025.12' diff --git a/array_api_compat/dask/array/__init__.py b/array_api_compat/dask/array/__init__.py index 1905c671..d25ae513 100644 --- a/array_api_compat/dask/array/__init__.py +++ b/array_api_compat/dask/array/__init__.py @@ -13,8 +13,8 @@ del Final # See the comment in the numpy __init__.py -__import__(__package__ + '.linalg') -__import__(__package__ + '.fft') +__import__(__spec__.parent + '.linalg') +__import__(__spec__.parent + '.fft') __all__ = sorted( set(__all__) diff --git a/array_api_compat/numpy/__init__.py b/array_api_compat/numpy/__init__.py index 81eaafef..bda4356f 100644 --- a/array_api_compat/numpy/__init__.py +++ b/array_api_compat/numpy/__init__.py @@ -20,9 +20,9 @@ # # It doesn't overwrite np.linalg from above. The import is generated # dynamically so that the library can be vendored. -__import__(__package__ + ".linalg") +__import__(__spec__.parent + ".linalg") -__import__(__package__ + ".fft") +__import__(__spec__.parent + ".fft") from .linalg import matrix_transpose, vecdot # type: ignore[no-redef] # noqa: F401 diff --git a/array_api_compat/torch/__init__.py b/array_api_compat/torch/__init__.py index 8263faa6..c5c801aa 100644 --- a/array_api_compat/torch/__init__.py +++ b/array_api_compat/torch/__init__.py @@ -10,8 +10,8 @@ from ._info import __array_namespace_info__ # noqa: F401 # See the comment in the numpy __init__.py -__import__(__package__ + '.linalg') -__import__(__package__ + '.fft') +__import__(__spec__.parent + '.linalg') +__import__(__spec__.parent + '.fft') __array_api_version__: Final = '2025.12'