Skip to content

Commit 867f2e3

Browse files
author
Matt Sokoloff
committed
fix mypy error
1 parent 1aaa7e8 commit 867f2e3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

labelbox/schema/model_run.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ def delete_model_run_data_rows(self, data_row_ids):
184184
def update_status(self,
185185
status: str,
186186
metadata: Optional[Dict[str, str]] = None,
187-
error_message: Optional[Dict[str, str]] = None):
187+
error_message: Optional[str] = None):
188188

189189
valid_statuses = [
190190
"EXPORTING_DATA", "PREPARING_DATA", "TRAINING_MODEL", "COMPLETE",
@@ -195,7 +195,7 @@ def update_status(self,
195195
f"Status must be one of : `{valid_statuses}`. Found : `{status}`"
196196
)
197197

198-
data = {'status': status}
198+
data: Dict[str, Any] = {'status': status}
199199
if error_message:
200200
data['errorMessage'] = error_message
201201

0 commit comments

Comments
 (0)