diff --git a/python/pyspark/sql/tests/pandas/test_pandas_map.py b/python/pyspark/sql/tests/pandas/test_pandas_map.py index ec9f208d08f9b..8f7229e1d74fd 100644 --- a/python/pyspark/sql/tests/pandas/test_pandas_map.py +++ b/python/pyspark/sql/tests/pandas/test_pandas_map.py @@ -31,7 +31,7 @@ pandas_requirement_message, pyarrow_requirement_message, ) -from pyspark.testing.utils import QuietTest +from pyspark.testing.utils import QuietTest, eventually if have_pandas: import pandas as pd @@ -381,6 +381,7 @@ def test_self_join(self): self.assertEqual(sorted(actual), sorted(expected)) # SPARK-33277 + @eventually(timeout=180, catch_assertions=True) def test_map_in_pandas_with_column_vector(self): path = tempfile.mkdtemp() shutil.rmtree(path)