fix

Author: WeichenXu123

python/pyspark/tests/test_rdd.py

@@ -834,9 +834,8 @@ def run_job(job_group, index):
             and then exits.
             """
             try:
-                self.sc.setJobGroup(job_group, "test rdd collect with setting job group")
-                self.sc.parallelize([15]).map(lambda x: time.sleep(x)).collect(
-                    job_group, "test rdd collect with setting job group")
+                self.sc.parallelize([15]).map(lambda x: time.sleep(x)) \
+                    .collectWithJobGroup(job_group, "test rdd collect with setting job group")
                 is_job_cancelled[index] = False
             except Exception:
                 # Assume that exception means job cancellation.