diff --git a/python/pyspark/sql/tests/test_udf_profiler.py b/python/pyspark/sql/tests/test_udf_profiler.py index 37f4a70fabd2..e6a7bf40b945 100644 --- a/python/pyspark/sql/tests/test_udf_profiler.py +++ b/python/pyspark/sql/tests/test_udf_profiler.py @@ -585,6 +585,7 @@ def summarize(left, right): for id in self.profile_results: self.assert_udf_profile_present(udf_id=id, expected_line_count_prefix=2) + @unittest.skipIf(not have_pyarrow, pyarrow_requirement_message) def test_perf_profiler_data_source(self): class TestDataSourceReader(DataSourceReader): def __init__(self, schema):