rtol=4 * 1e-4, atol=2 * 1e-4

microsoft · Aug 10, 2024 · f3aecaf · f3aecaf
1 parent 33f8f3f
commit f3aecaf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_sklearn_decomposition.py b/tests/test_sklearn_decomposition.py
@@ -32,7 +32,7 @@ def _fit_model_pca(self, model, precompute=False):
 
         torch_model = hummingbird.ml.convert(model, "torch")
         self.assertTrue(torch_model is not None)
-        np.testing.assert_allclose(model.transform(X_test), torch_model.transform(X_test), rtol=1e-6, atol=1e-4)
+        np.testing.assert_allclose(model.transform(X_test), torch_model.transform(X_test), rtol=4 * 1e-4, atol=2 * 1e-4)
 
     # PCA n_components none
     def test_pca_converter_none(self):