cleanup

Author: Matt Sokoloff

labelbox/schema/model_run.py

@@ -199,7 +199,7 @@ def delete_model_run_data_rows(self, data_row_ids: List[str]):
     def assign_data_rows_to_split(self,
                                   data_row_ids: List[str],
                                   split: Union[DataSplit, str],
-                                  timeout_seconds=60):
+                                  timeout_seconds=120):
 
         split_value = split.value if isinstance(split, DataSplit) else split
 
@@ -212,7 +212,7 @@ def assign_data_rows_to_split(self,
 
         if split_value not in valid_splits:
             raise ValueError(
-                f"split must be one of : `{valid_splits}`. Found : `{split}`")
+                f"`split` must be one of : `{valid_splits}`. Found : `{split}`")
 
         task_id = self.client.execute(
             """mutation assignDataSplitPyApi($modelRunId: ID!, $data: CreateAssignDataRowsToDataSplitTaskInput!){