From 7e932eca3141aa44a13ed2aa46f608fd31e2c7da Mon Sep 17 00:00:00 2001 From: fxmarty <9808326+fxmarty@users.noreply.github.com> Date: Wed, 23 Aug 2023 01:39:40 +0900 Subject: [PATCH] Fix staging tests following transformers 4.32 release (#1309) and some more breaking changes --- tests/test_configuration_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_configuration_utils.py b/tests/test_configuration_utils.py index 8706df6c7e..4c721f089d 100644 --- a/tests/test_configuration_utils.py +++ b/tests/test_configuration_utils.py @@ -71,7 +71,7 @@ def test_push_to_hub(self): config = FakeConfig(attribute=15) with tempfile.TemporaryDirectory() as tmp_dir: config.save_pretrained( - os.path.join(tmp_dir, "optimum-test-base-config"), push_to_hub=True, use_auth_token=self._token + os.path.join(tmp_dir, "optimum-test-base-config"), push_to_hub=True, token=self._token ) new_config = FakeConfig.from_pretrained(f"{USER}/optimum-test-base-config") @@ -86,7 +86,7 @@ def test_push_to_hub_in_organization(self): config.save_pretrained( os.path.join(tmp_dir, "optimum-test-base-config-org"), push_to_hub=True, - use_auth_token=self._token, + token=self._token, organization="valid_org", )