diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py
index a0f7f54e65213ccc7cc95237a6228ff3a21a2188..e12abeba019f755ebc350e869269f7be0c70893e 100644
--- a/python/pyspark/ml/regression.py
+++ b/python/pyspark/ml/regression.py
@@ -701,11 +701,7 @@ class AFTSurvivalRegression(JavaEstimator, HasFeaturesCol, HasLabelCol, HasPredi
                   quantilesCol=None):
         """
         kwargs = self.setParams._input_kwargs
-        if quantileProbabilities is None:
-            return self._set(**kwargs).setQuantileProbabilities([0.01, 0.05, 0.1, 0.25, 0.5,
-                                                                 0.75, 0.9, 0.95, 0.99])
-        else:
-            return self._set(**kwargs)
+        return self._set(**kwargs)
 
     def _create_model(self, java_model):
         return AFTSurvivalRegressionModel(java_model)