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

[model upload] Support for organizations

Browse Source
This commit is contained in:
Julien Chaumond
2020-03-09 17:29:49 -04:00
parent 525b6b1c54
commit cbf8f5d32b
3 changed files with 50 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_hf_api.py
View File

@@ -67,8 +67,17 @@ class HfApiEndpointsTest(HfApiCommonTest):
cls._api.delete_obj(token=cls._token, filename=FILE_KEY)
def test_whoami(self):
user = self._api.whoami(token=self._token)
user, orgs = self._api.whoami(token=self._token)
self.assertEqual(user, USER)
self.assertIsInstance(orgs, list)
def test_presign_invalid_org(self):
with self.assertRaises(HTTPError):
_ = self._api.presign(token=self._token, filename="fake_org.txt", organization="fake")
def test_presign_valid_org(self):
urls = self._api.presign(token=self._token, filename="valid_org.txt", organization="valid_org")
self.assertIsInstance(urls, PresignedUrl)
def test_presign(self):
for FILE_KEY, FILE_PATH in FILES: