diff --git a/databricks/koalas/series.py b/databricks/koalas/series.py index f6e011830d..64c3b0aa54 100644 --- a/databricks/koalas/series.py +++ b/databricks/koalas/series.py @@ -1478,7 +1478,11 @@ def to_list(self): to be small, as all the data is loaded into the driver's memory. """ - return self._to_internal_pandas().to_list() + # pandas<0.24 doesn't support `to_list()` as an alias for `tolist()` + if LooseVersion(pd.__version__) < LooseVersion("0.24"): + return self._to_internal_pandas().tolist() + else: + return self._to_internal_pandas().to_list() tolist = to_list diff --git a/databricks/koalas/tests/test_series.py b/databricks/koalas/tests/test_series.py index 9aa29f59d6..56248e56db 100644 --- a/databricks/koalas/tests/test_series.py +++ b/databricks/koalas/tests/test_series.py @@ -861,7 +861,9 @@ def test_is_unique(self): def test_to_list(self): if LooseVersion(pd.__version__) >= LooseVersion("0.24.0"): - self.assertEqual(self.kser.to_list(), self.pser.to_list()) + self.assert_eq(self.kser.to_list(), self.pser.to_list()) + else: + self.assert_eq(self.kser.tolist(), self.pser.tolist()) def test_append(self): pser1 = pd.Series([1, 2, 3], name="0")