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

Remove six.

Browse Source
This commit is contained in:
Aymeric Augustin
2019-12-22 17:56:09 +01:00
parent 6b2200fc88
commit 8af25b1664
19 changed files with 61 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_hf_api.py
View File

@@ -19,7 +19,6 @@ import time
import unittest
import requests
import six
from requests.exceptions import HTTPError
from transformers.hf_api import HfApi, HfFolder, PresignedUrl, S3Obj
@@ -50,7 +49,7 @@ class HfApiLoginTest(HfApiCommonTest):
def test_login_valid(self):
token = self._api.login(username=USER, password=PASS)
self.assertIsInstance(token, six.string_types)
self.assertIsInstance(token, str)
class HfApiEndpointsTest(HfApiCommonTest):
@@ -74,7 +73,7 @@ class HfApiEndpointsTest(HfApiCommonTest):
def test_presign_and_upload(self):
for FILE_KEY, FILE_PATH in FILES:
access_url = self._api.presign_and_upload(token=self._token, filename=FILE_KEY, filepath=FILE_PATH)
self.assertIsInstance(access_url, six.string_types)
self.assertIsInstance(access_url, str)
with open(FILE_PATH, "r") as f:
body = f.read()
r = requests.get(access_url)