From 979facc48375de3091acde8ebd804a80d6852328 Mon Sep 17 00:00:00 2001 From: wojciechwoszczek Date: Wed, 6 Dec 2023 19:20:24 +0100 Subject: [PATCH] fix test --- .github/workflows/backend.yml | 2 +- tests/test_train_model.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/backend.yml b/.github/workflows/backend.yml index f05a51b..0c727f3 100644 --- a/.github/workflows/backend.yml +++ b/.github/workflows/backend.yml @@ -45,7 +45,7 @@ run: | pytest tests/test_mlflow.py pytest tests/test_train_model.py - continue-on-error: true + continue-on-error: false cd_push_to_dockerhub: diff --git a/tests/test_train_model.py b/tests/test_train_model.py index e61842d..3e532f8 100644 --- a/tests/test_train_model.py +++ b/tests/test_train_model.py @@ -33,7 +33,7 @@ def test_epoch_loss(): def test_accuracy_threshold(): for index, row in runs_final.iterrows(): - assert row["metrics.test_accuracy"]>70, "Model accuracy under threshold, review model architecture" + assert row["metrics.validation_accuracy"]>70, "Model accuracy under threshold, review model architecture" def test_artifacts(): for index, row in runs_final.iterrows():