From 41cae240279f48f46a895f81a6698626e75cd0d9 Mon Sep 17 00:00:00 2001 From: Dongjoon Hyun Date: Thu, 2 Dec 2021 09:46:30 -0800 Subject: [PATCH] [SPARK-37522][PYTHON][TESTS] Fix MultilayerPerceptronClassifierTest.test_raw_and_probability_prediction --- python/pyspark/ml/tests/test_algorithms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):