Fix test for file not found (#20604)

5764efe5 · Sylvain Gugger · GitHub · 720e9599 · 5764efe5
Unverified Commit 5764efe5 authored Dec 05, 2022 by Sylvain Gugger Committed by GitHub Dec 05, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/models/auto/test_modeling_tf_auto.py tests/models/auto/test_modeling_tf_auto.py +1 -1

No files found.
--- a/tests/models/auto/test_modeling_tf_auto.py
+++ b/tests/models/auto/test_modeling_tf_auto.py
@@ -281,7 +281,7 @@ class TFAutoModelTest(unittest.TestCase):
    def test_model_file_not_found(self):
        with self.assertRaisesRegex(
            EnvironmentError,
-            "hf-internal-testing/config-no-model does not appear to have a file named tf_model.h5",
+            "hf-internal-testing/config-no-model does not appear to have a file named pytorch_model.bin",
        ):
            _ = TFAutoModel.from_pretrained("hf-internal-testing/config-no-model")