From 5764efe544de909e93bfde16489b5a0975fce1c2 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Mon, 5 Dec 2022 18:33:56 -0500 Subject: [PATCH] Fix test for file not found (#20604) --- 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 2b4b625e23..1a355d88bb 100644 --- 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")