From 4bd5d158e66e6b1ca5f1a767738f0674f0dc8095 Mon Sep 17 00:00:00 2001 From: Aaron Meurer Date: Tue, 12 Jan 2021 16:21:46 -0700 Subject: Use "import numpy as np" in the array_api submodule This avoids importing everything inside the individual functions, but still is preferred over importing the functions used explicitly, as most of them clash with the wrapper function names. --- numpy/_array_api/_statistical_functions.py | 23 +++++++++-------------- 1 file changed, 9 insertions(+), 14 deletions(-) (limited to 'numpy/_array_api/_statistical_functions.py') diff --git a/numpy/_array_api/_statistical_functions.py b/numpy/_array_api/_statistical_functions.py index 339835095..833c47f66 100644 --- a/numpy/_array_api/_statistical_functions.py +++ b/numpy/_array_api/_statistical_functions.py @@ -1,29 +1,24 @@ +import numpy as np + def max(x, /, *, axis=None, keepdims=False): - from .. import max - return max(x, axis=axis, keepdims=keepdims) + return np.max(x, axis=axis, keepdims=keepdims) def mean(x, /, *, axis=None, keepdims=False): - from .. import mean - return mean(x, axis=axis, keepdims=keepdims) + return np.mean(x, axis=axis, keepdims=keepdims) def min(x, /, *, axis=None, keepdims=False): - from .. import min - return min(x, axis=axis, keepdims=keepdims) + return np.min(x, axis=axis, keepdims=keepdims) def prod(x, /, *, axis=None, keepdims=False): - from .. import prod - return prod(x, axis=axis, keepdims=keepdims) + return np.prod(x, axis=axis, keepdims=keepdims) def std(x, /, *, axis=None, correction=0.0, keepdims=False): - from .. import std # Note: the keyword argument correction is different here - return std(x, axis=axis, ddof=correction, keepdims=keepdims) + return np.std(x, axis=axis, ddof=correction, keepdims=keepdims) def sum(x, /, *, axis=None, keepdims=False): - from .. import sum - return sum(x, axis=axis, keepdims=keepdims) + return np.sum(x, axis=axis, keepdims=keepdims) def var(x, /, *, axis=None, correction=0.0, keepdims=False): - from .. import var # Note: the keyword argument correction is different here - return var(x, axis=axis, ddof=correction, keepdims=keepdims) + return np.var(x, axis=axis, ddof=correction, keepdims=keepdims) -- cgit v1.2.1