From 9db01a4ee74b5fed75e6207ef5b648e59c710622 Mon Sep 17 00:00:00 2001 From: Michael Waskom Date: Wed, 7 Feb 2024 13:56:59 -0500 Subject: [PATCH] Fix direction of loss assertion --- ci/check_loss.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/ci/check_loss.py b/ci/check_loss.py index 57928f3..848b1f3 100644 --- a/ci/check_loss.py +++ b/ci/check_loss.py @@ -24,9 +24,8 @@ results_text = m.group(1).strip().replace(" ", "") results = pd.read_table(StringIO(results_text), sep="|") - print(results) # TODO debugging train_loss = float(results["TrainingLoss"].iloc[-1]) val_loss = float(results["ValidationLoss"].iloc[-1]) - print("Loss: {train_loss:.2f} (training), {val_loss:.2f} (validation)") - sys.exit(val_loss < 0.25) # Arbitrary threshold + print(f"Loss: {train_loss:.2f} (training), {val_loss:.2f} (validation)") + sys.exit(val_loss > 0.25) # Arbitrary threshold