[cuda ext tests] fixing tests (#11619)

* fixing tests * cleanup
2021-05-06 13:35:28 -07:00
parent 44c5621db0
commit 619200cc42
3 changed files with 8 additions and 5 deletions
--- a/tests/extended/test_trainer_ext.py
+++ b/tests/extended/test_trainer_ext.py
@@ -167,8 +167,8 @@ class TestTrainerExt(TestCasePlus):
        # test if do_predict saves generations and metrics
        contents = os.listdir(output_dir)
        contents = {os.path.basename(p) for p in contents}
-        assert "test_generations.txt" in contents
-        assert "test_results.json" in contents
+        assert "generated_predictions.txt" in contents
+        assert "predict_results.json" in contents

    def run_trainer(
        self,