From 5009b54eede085f203f465529b52f9417db2bbfc Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Mon, 5 Dec 2022 18:10:03 -0500 Subject: [PATCH] Fix test for file not found --- tests/models/auto/test_modeling_tf_auto.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/auto/test_modeling_tf_auto.py b/tests/models/auto/test_modeling_tf_auto.py index 2b4b625e2305..1a355d88bb5a 100644 --- a/tests/models/auto/test_modeling_tf_auto.py +++ b/tests/models/auto/test_modeling_tf_auto.py @@ -281,7 +281,7 @@ def test_revision_not_found(self): 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")