From a735f727cc93f96abc5b39e776716891667bb565 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Tue, 23 Mar 2021 19:03:06 -0400 Subject: [PATCH] Fix test_trainer_distributed (#10875) --- tests/test_trainer_distributed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_trainer_distributed.py b/tests/test_trainer_distributed.py index d6783a6281..d892745968 100644 --- a/tests/test_trainer_distributed.py +++ b/tests/test_trainer_distributed.py @@ -119,7 +119,7 @@ if __name__ == "__main__": p = trainer.predict(dataset) logger.info(p.metrics) - if p.metrics["eval_success"] is not True: + if p.metrics["test_success"] is not True: logger.error(p.metrics) exit(1) @@ -133,7 +133,7 @@ if __name__ == "__main__": p = trainer.predict(dataset) logger.info(p.metrics) - if p.metrics["eval_success"] is not True: + if p.metrics["test_success"] is not True: logger.error(p.metrics) exit(1)