Skip to content

Commit

Permalink
and some more breaking changes
Browse files Browse the repository at this point in the history
  • Loading branch information
fxmarty committed Aug 22, 2023
1 parent f600bc6 commit 2e707af
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_configuration_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand All @@ -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",
)

Expand Down

0 comments on commit 2e707af

Please sign in to comment.