diff --git a/python/pyspark/sql/session.py b/python/pyspark/sql/session.py index 0cc2d7d3f13e0..0cb02b22a60d0 100644 --- a/python/pyspark/sql/session.py +++ b/python/pyspark/sql/session.py @@ -2048,7 +2048,7 @@ def registerProgressHandler(self, handler: "ProgressHandler") -> None: >>> def progress_handler(stages, inflight_tasks, done): ... print(f"{len(stages)} Stages known, Done: {done}") >>> spark.registerProgressHandler(progress_handler) - >>> res = spark.range(10).repartition(1).collect() + >>> res = spark.range(10).repartition(1).collect() # doctest: +SKIP 3 Stages known, Done: False 3 Stages known, Done: True >>> spark.clearProgressHandlers()