SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Prepare transformers for v0.8.0 huggingface-hub release (#17716)

Browse Source 
* Prepare CI for v0.8.0

* pin hfh (revert before merge)

* Revert "pin hfh (revert before merge)"

This reverts commit a0103140e1c77b810ffcb735192968bc03be3e1f.

* Test rc3

* Test latest rc

* Unpin to the RC

Co-authored-by: Sylvain Gugger <Sylvain.gugger@gmail.com>
This commit is contained in:
Lysandre Debut
2022-06-21 17:51:18 +02:00
committed by GitHub
parent 7bc88c0511
commit 6a5272b205
11 changed files with 75 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/trainer/test_trainer.py
View File

@@ -30,7 +30,7 @@ from unittest.mock import Mock, patch
import numpy as np
from huggingface_hub import Repository, delete_repo, login
from huggingface_hub import HfFolder, Repository, delete_repo, set_access_token
from parameterized import parameterized
from requests.exceptions import HTTPError
from transformers import (
@@ -43,7 +43,7 @@ from transformers import (
)
from transformers.testing_utils import (
ENDPOINT_STAGING,
PASS,
TOKEN,
USER,
CaptureLogger,
TestCasePlus,
@@ -1966,18 +1966,20 @@ class TrainerIntegrationTest(TestCasePlus, TrainerIntegrationCommon):
class TrainerIntegrationWithHubTester(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls._token = login(username=USER, password=PASS)
cls._token = TOKEN
set_access_token(TOKEN)
HfFolder.save_token(TOKEN)
@classmethod
def tearDownClass(cls):
for model in ["test-trainer", "test-trainer-epoch", "test-trainer-step"]:
try:
delete_repo(token=cls._token, name=model)
delete_repo(token=cls._token, repo_id=model)
except HTTPError:
pass
try:
delete_repo(token=cls._token, name="test-trainer-org", organization="valid_org")
delete_repo(token=cls._token, repo_id="valid_org/test-trainer-org")
except HTTPError:
pass