diff --git a/python/pyspark/ml/tests/test_algorithms.py b/python/pyspark/ml/tests/test_algorithms.py index d1a5dea7693a0..bf74988a7c097 100644 --- a/python/pyspark/ml/tests/test_algorithms.py +++ b/python/pyspark/ml/tests/test_algorithms.py @@ -101,7 +101,7 @@ def test_raw_and_probability_prediction(self): expected_rawPrediction = [-11.6081922998, -8.15827998691, 22.17757045] self.assertTrue(result.prediction, expected_prediction) self.assertTrue(np.allclose(result.probability, expected_probability, atol=1e-4)) - self.assertTrue(np.allclose(result.rawPrediction, expected_rawPrediction, rtol=0.1)) + self.assertTrue(np.allclose(result.rawPrediction, expected_rawPrediction, rtol=0.11)) class OneVsRestTests(SparkSessionTestCase):