Fix test_trainer_distributed (#10875)

a735f727 · Sylvain Gugger · GitHub · 8c297cdb · a735f727
Unverified Commit a735f727 authored Mar 23, 2021 by Sylvain Gugger Committed by GitHub Mar 23, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/test_trainer_distributed.py tests/test_trainer_distributed.py +2 -2

No files found.
--- 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)