From af3394d10dcbf434238b7262f39bd865d0543bef Mon Sep 17 00:00:00 2001 From: jbrockmendel Date: Wed, 18 Mar 2020 10:19:44 -0700 Subject: [PATCH] TYP: annotate to_numpy --- pandas/core/arrays/base.py | 4 +++- pandas/core/arrays/numpy_.py | 5 ++++- pandas/core/frame.py | 2 +- pandas/tests/extension/decimal/array.py | 4 +++- 4 files changed, 11 insertions(+), 4 deletions(-) diff --git a/pandas/core/arrays/base.py b/pandas/core/arrays/base.py index 6aa303dd04703..c42c1539daa5a 100644 --- a/pandas/core/arrays/base.py +++ b/pandas/core/arrays/base.py @@ -356,7 +356,9 @@ def __iter__(self): for i in range(len(self)): yield self[i] - def to_numpy(self, dtype=None, copy=False, na_value=lib.no_default): + def to_numpy( + self, dtype=None, copy: bool = False, na_value=lib.no_default + ) -> np.ndarray: """ Convert to a NumPy ndarray. diff --git a/pandas/core/arrays/numpy_.py b/pandas/core/arrays/numpy_.py index 0e64967ce93a6..e8333606ec54c 100644 --- a/pandas/core/arrays/numpy_.py +++ b/pandas/core/arrays/numpy_.py @@ -435,7 +435,10 @@ def skew(self, axis=None, dtype=None, out=None, keepdims=False, skipna=True): # ------------------------------------------------------------------------ # Additional Methods - def to_numpy(self, dtype=None, copy=False, na_value=lib.no_default): + + def to_numpy( + self, dtype=None, copy: bool = False, na_value=lib.no_default + ) -> np.ndarray: result = np.asarray(self._ndarray, dtype=dtype) if (copy or na_value is not lib.no_default) and result is self._ndarray: diff --git a/pandas/core/frame.py b/pandas/core/frame.py index baa6fb07ff233..b9e43b1cd9b05 100644 --- a/pandas/core/frame.py +++ b/pandas/core/frame.py @@ -1253,7 +1253,7 @@ def from_dict(cls, data, orient="columns", dtype=None, columns=None) -> "DataFra return cls(data, index=index, columns=columns, dtype=dtype) - def to_numpy(self, dtype=None, copy=False) -> np.ndarray: + def to_numpy(self, dtype=None, copy: bool = False) -> np.ndarray: """ Convert the DataFrame to a NumPy array. diff --git a/pandas/tests/extension/decimal/array.py b/pandas/tests/extension/decimal/array.py index 9384ed5199c1f..85d8ad6ec6e38 100644 --- a/pandas/tests/extension/decimal/array.py +++ b/pandas/tests/extension/decimal/array.py @@ -79,7 +79,9 @@ def _from_factorized(cls, values, original): _HANDLED_TYPES = (decimal.Decimal, numbers.Number, np.ndarray) - def to_numpy(self, dtype=None, copy=False, na_value=no_default, decimals=None): + def to_numpy( + self, dtype=None, copy: bool = False, na_value=no_default, decimals=None + ) -> np.ndarray: result = np.asarray(self, dtype=dtype) if decimals is not None: result = np.asarray([round(x, decimals) for x in result])