Lower check precision (mlflow#896)

rajib2k5 · Feb 14, 2019 · d6f8612 · d6f8612
1 parent 5099a01
commit d6f8612
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/keras/test_keras_model_export.py b/tests/keras/test_keras_model_export.py
@@ -92,7 +92,7 @@ def test_model_save_load(model, model_path, data, predicted):
                                          pandas_df=pd.DataFrame(x),
                                          result_type="float")
     np.testing.assert_array_almost_equal(
-        np.array(spark_udf_preds), predicted.reshape(len(spark_udf_preds)), decimal=6)
+        np.array(spark_udf_preds), predicted.reshape(len(spark_udf_preds)), decimal=4)
 
 
 def test_model_log(tracking_uri_mock, model, data, predicted):  # pylint: disable=unused-argument