Skip to content

Commit 3df91fd

Browse files
author
Val Brodsky
committed
A more full-proof way to handle unknown exceptions for model setup complete
1 parent 20cfe48 commit 3df91fd

File tree

2 files changed

+14
-2
lines changed

2 files changed

+14
-2
lines changed

libs/labelbox/src/labelbox/schema/project.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1272,7 +1272,13 @@ def add_model_config(self, model_config_id: str) -> str:
12721272
try:
12731273
result = self.client.execute(query, params)
12741274
except LabelboxError as e:
1275-
error_content = _error_message_for_unparsed_graphql_error(e.message)
1275+
if e.message.startswith(
1276+
"Unknown error: "
1277+
): # unfortunate hack to handle unparsed graphql errors
1278+
error_content = _error_message_for_unparsed_graphql_error(
1279+
e.message)
1280+
else:
1281+
error_content = e.message
12761282
raise LabelboxError(message=error_content) from e
12771283

12781284
if not result:

libs/labelbox/src/labelbox/schema/project_model_config.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,13 @@ def delete(self) -> bool:
3535
try:
3636
result = self.client.execute(query, params)
3737
except LabelboxError as e:
38-
error_content = _error_message_for_unparsed_graphql_error(e.message)
38+
if e.message.startswith(
39+
"Unknown error: "
40+
): # unfortunate hack to handle unparsed graphql errors
41+
error_content = _error_message_for_unparsed_graphql_error(
42+
e.message)
43+
else:
44+
error_content = e.message
3945
raise LabelboxError(message=error_content) from e
4046

4147
return result["deleteProjectModelConfig"]["success"]

0 commit comments

Comments
 (0)