Fix staging tests following transformers 4.32 release (#1309)

and some more breaking changes
huggingface · Aug 22, 2023 · 7e932ec · 7e932ec
1 parent d99a418
commit 7e932ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",
             )