From 6479f493ed308fa35e92fd6acbe0606680bd546e Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 09:39:43 -0400 Subject: [PATCH] updated reqs --- tests/test_models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index 79dfaf34..e0b0a36e 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -46,7 +46,8 @@ def clear_tt_dir(): def assert_ok_acc(trainer): # this model should get 0.80+ acc - assert trainer.tng_tqdm_dic['val_acc'] > 0.70, "model failed to get expected 0.80 validation accuracy" + acc = trainer.tng_tqdm_dic['val_acc'] + assert acc > 0.70, f'model failed to get expected 0.80 validation accuracy. Got: {acc}' def test_cpu_model():