Include model_group_id upon model deployment

openvinotoolkit · Jun 18, 2024 · 94854ab · 94854ab
1 parent 34eb5f5
commit 94854ab
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/geti_sdk/rest_clients/deployment_client.py b/geti_sdk/rest_clients/deployment_client.py
@@ -361,9 +361,11 @@ def _backend_deploy_project(
         # Fetch the deployment package
         deployment = self._fetch_deployment(deployment_id=code_deployment.id)
 
-        # Attach the appropriate hyper parameters to the deployed models
+        # Attach the appropriate hyper parameters and model_group_id to the deployed
+        # models
         for index, model in enumerate(deployment.models):
             model.hyper_parameters = hyper_parameters[index]
+            model.model_group_id = sorted_optimized_models[index].model_group_id
 
         # Save the deployment, if needed
         if output_folder is not None: