Skip to content

Commit a083993

Browse files
rllinrllin
authored andcommitted
catch actual errors
1 parent c656f3c commit a083993

File tree

2 files changed

+18
-1
lines changed

2 files changed

+18
-1
lines changed

labelbox/client.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,18 @@ def convert_value(value):
120120
raise labelbox.exceptions.LabelboxError(
121121
"Unknown error during Client.query(): " + str(e), e)
122122

123+
# if we do return a proper error code, reraise
124+
if response.status_code != requests.codes.ok:
125+
message = f"{response.status_code} {response.reason}"
126+
cause = None
127+
try:
128+
r_json = response.json()
129+
except:
130+
pass
131+
else:
132+
cause = r_json.get('message')
133+
raise labelbox.exceptions.LabelboxError(message, cause)
134+
123135
try:
124136
response = response.json()
125137
except:

setup.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,12 @@
1313
long_description_content_type="text/markdown",
1414
url="https://labelbox.com",
1515
packages=setuptools.find_packages(),
16-
install_requires=["backoff==1.10.0", "ndjson==0.3.1", "requests==2.22.0"],
16+
install_requires=[
17+
"backoff==1.10.0",
18+
"ndjson==0.3.1",
19+
"requests==2.22.0",
20+
"google-api-core>=1.22.1",
21+
],
1722
classifiers=[
1823
'Development Status :: 3 - Alpha',
1924
'License :: OSI Approved :: Apache Software License',

0 commit comments

Comments
 (0)