Skip to content

Commit abf4727

Browse files
committed
Update workflows for TensorFlow/Keras 2.14
Signed-off-by: Beat Buesser <[email protected]>
1 parent 709adc3 commit abf4727

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/estimators/classification/test_scikitlearn.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ def test_type(self):
269269
def test_predict(self):
270270
y_predicted = self.classifier.predict(self.x_test_iris[0:1])
271271
y_expected = np.asarray([[0.07997696, 0.36272544, 0.5572976]])
272-
np.testing.assert_array_almost_equal(y_predicted, y_expected, decimal=4)
272+
np.testing.assert_array_almost_equal(y_predicted, y_expected, decimal=3)
273273

274274
def test_class_gradient_none_1(self):
275275
grad_predicted = self.classifier.class_gradient(self.x_test_iris[0:1], label=None)
@@ -280,7 +280,7 @@ def test_class_gradient_none_1(self):
280280
[0.6508137, 0.26377308, 1.54522324, 0.80972391],
281281
]
282282
]
283-
np.testing.assert_array_almost_equal(grad_predicted, grad_expected, decimal=4)
283+
np.testing.assert_array_almost_equal(grad_predicted, grad_expected, decimal=3)
284284

285285
def test_class_gradient_none_2(self):
286286
grad_predicted = self.classifier.class_gradient(self.x_test_iris[0:2], label=None)
@@ -296,7 +296,7 @@ def test_class_gradient_none_2(self):
296296
[0.70875132, 0.25104877, 1.70929277, 0.88410652],
297297
],
298298
]
299-
np.testing.assert_array_almost_equal(grad_predicted, grad_expected, decimal=4)
299+
np.testing.assert_array_almost_equal(grad_predicted, grad_expected, decimal=3)
300300

301301
def test_class_gradient_int_1(self):
302302
grad_predicted = self.classifier.class_gradient(self.x_test_iris[0:1], label=1)
@@ -326,7 +326,7 @@ def test_class_gradient_list_2(self):
326326
[[-0.56317306, -0.70493776, -0.98908573, -0.67106259]],
327327
[[0.70866591, 0.25158876, 1.70947325, 0.88450021]],
328328
]
329-
np.testing.assert_array_almost_equal(grad_predicted, grad_expected, decimal=4)
329+
np.testing.assert_array_almost_equal(grad_predicted, grad_expected, decimal=3)
330330

331331
def test_class_gradient_label_wrong_type(self):
332332

0 commit comments

Comments
 (0)