Merge remote-tracking branch 'origin/re-design-sdk' into re-design-sdk

Author: Vaghinak Basentsyan

src/superannotate/lib/app/interface/sdk_interface.py

@@ -2697,7 +2697,18 @@ def stop_model_training(model):
     :return: the metadata of the now, stopped model
     :rtype: dict
     """
-    response = controller.stop_model_training(model["id"])
+
+    model_id = None
+    if isinstance(model, dict):
+        model_id = model['id']
+    else:
+        res = controller.search_models(name=model).data
+        if len(res):
+            model_id = res[0]['id']
+        else:
+            raise AppException("Model not found.")
+
+    response = controller.stop_model_training(model_id=model_id)
 
     if not response.errors:
         logger.info("Stopped model training")

src/superannotate/lib/infrastructure/services.py

@@ -158,7 +158,7 @@ class SuperannotateBackendService(BaseBackendService):
     URL_GET_TEMPLATES = "templates"
     URL_PROJECT_WORKFLOW_ATTRIBUTE = "project/{}/workflow_attribute"
     URL_MODELS = "ml_models"
-    URL_STOP_MODEL_TRAINING = "ml_models/{}/stopTrainingJob"
+    URL_STOP_MODEL_TRAINING = "ml_model/{}/stopTrainingJob"
     URL_GET_MODEL_METRICS = "ml_models/{}/getCurrentMetrics"
     URL_BULK_GET_FOLDERS = "foldersByTeam"
     URL_GET_EXPORT = "export/{}"