From 2b3ece5d375274965a4a98e914dfb9d3cefd98d1 Mon Sep 17 00:00:00 2001 From: Hyukjin Kwon Date: Thu, 24 Feb 2022 13:02:24 +0900 Subject: [PATCH] Revert "[SPARK-38297][PYTHON] Explicitly cast the return value at DataFrame.to_numpy in POS" This reverts commit a0d2be565486367abd6b637c98634c35420994ce. --- python/pyspark/pandas/generic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/pyspark/pandas/generic.py b/python/pyspark/pandas/generic.py index c26b516c59be2..cdd8f67213f1b 100644 --- a/python/pyspark/pandas/generic.py +++ b/python/pyspark/pandas/generic.py @@ -573,7 +573,7 @@ def to_numpy(self) -> np.ndarray: >>> ps.Series(['a', 'b', 'a']).to_numpy() array(['a', 'b', 'a'], dtype=object) """ - return cast(np.ndarray, self._to_pandas().values) + return self.to_pandas().values @property def values(self) -> np.ndarray: