Restore proper import for HTTPError.

7c681264 · Aymeric Augustin · 939148b0 · 7c681264 · 7c681264
Commit 7c681264 authored Dec 21, 2019 by Aymeric Augustin
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

transformers/commands/user.py transformers/commands/user.py +3 -1

transformers/tests/hf_api_test.py transformers/tests/hf_api_test.py +2 -1

No files found.
--- a/transformers/commands/user.py
+++ b/transformers/commands/user.py
@@ -3,8 +3,10 @@ from argparse import ArgumentParser
 from getpass import getpass
 from typing import List, Union

+from requests.exceptions import HTTPError
+
 from transformers.commands import BaseTransformersCLICommand
-from transformers.hf_api import HfApi, HfFolder, HTTPError
+from transformers.hf_api import HfApi, HfFolder


 class UserCommands(BaseTransformersCLICommand):

--- a/transformers/tests/hf_api_test.py
+++ b/transformers/tests/hf_api_test.py
@@ -20,8 +20,9 @@ import unittest

 import requests
 import six
+from requests.exceptions import HTTPError

-from transformers.hf_api import HfApi, HfFolder, HTTPError, PresignedUrl, S3Obj
+from transformers.hf_api import HfApi, HfFolder, PresignedUrl, S3Obj


 USER = "__DUMMY_TRANSFORMERS_USER__"