From 4f302fe45fc501aab59f88b6f2aaabb42c661478 Mon Sep 17 00:00:00 2001 From: Beat Buesser Date: Thu, 16 Nov 2023 23:48:43 +0100 Subject: [PATCH] Update workflows for TensorFlow/Keras 2.14 Signed-off-by: Beat Buesser --- tests/estimators/classification/test_scikitlearn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/estimators/classification/test_scikitlearn.py b/tests/estimators/classification/test_scikitlearn.py index 56ae97f9e7..14fccaf1c5 100644 --- a/tests/estimators/classification/test_scikitlearn.py +++ b/tests/estimators/classification/test_scikitlearn.py @@ -315,7 +315,7 @@ def test_class_gradient_int_2(self): def test_class_gradient_list_1(self): grad_predicted = self.classifier.class_gradient(self.x_test_iris[0:1], label=[1]) - grad_expected = [[[-0.56317311, -0.70493763, -0.98874801, -0.67053026]]] + grad_expected = [[[-0.56317311, -0.70493763, -0.98874801, -0.67106276]]] for i_shape in range(4): self.assertAlmostEqual(grad_predicted[0, 0, i_shape], grad_expected[0][0][i_shape], 3)