From 9578636259f86267c2253f4af2510ce1eeaf084c Mon Sep 17 00:00:00 2001 From: Aaron Meurer Date: Tue, 12 Jan 2021 12:34:59 -0700 Subject: Make the array_api submodules private, and remove __all__ from individual files The specific submodule organization is an implementation detail and should not be used. Only the top-level numpy._array_api namespace should be used. --- numpy/_array_api/_creation_functions.py | 76 +++++++++++++++++++++++++++++++++ 1 file changed, 76 insertions(+) create mode 100644 numpy/_array_api/_creation_functions.py (limited to 'numpy/_array_api/_creation_functions.py') diff --git a/numpy/_array_api/_creation_functions.py b/numpy/_array_api/_creation_functions.py new file mode 100644 index 000000000..4e91aa443 --- /dev/null +++ b/numpy/_array_api/_creation_functions.py @@ -0,0 +1,76 @@ +def arange(start, /, *, stop=None, step=1, dtype=None, device=None): + from .. import arange + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return arange(start, stop=stop, step=step, dtype=dtype) + +def empty(shape, /, *, dtype=None, device=None): + from .. import empty + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return empty(shape, dtype=dtype) + +def empty_like(x, /, *, dtype=None, device=None): + from .. import empty_like + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return empty_like(x, dtype=dtype) + +def eye(N, /, *, M=None, k=0, dtype=None, device=None): + from .. import eye + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return eye(N, M=M, k=k, dtype=dtype) + +def full(shape, fill_value, /, *, dtype=None, device=None): + from .. import full + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return full(shape, fill_value, dtype=dtype) + +def full_like(x, fill_value, /, *, dtype=None, device=None): + from .. import full_like + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return full_like(x, fill_value, dtype=dtype) + +def linspace(start, stop, num, /, *, dtype=None, device=None, endpoint=True): + from .. import linspace + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return linspace(start, stop, num, dtype=dtype, endpoint=endpoint) + +def ones(shape, /, *, dtype=None, device=None): + from .. import ones + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return ones(shape, dtype=dtype) + +def ones_like(x, /, *, dtype=None, device=None): + from .. import ones_like + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return ones_like(x, dtype=dtype) + +def zeros(shape, /, *, dtype=None, device=None): + from .. import zeros + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return zeros(shape, dtype=dtype) + +def zeros_like(x, /, *, dtype=None, device=None): + from .. import zeros_like + if device is not None: + # Note: Device support is not yet implemented on ndarray + raise NotImplementedError("Device support is not yet implemented") + return zeros_like(x, dtype=dtype) -- cgit v1.2.1