From 1ad937143e03a6fc04f1c70a907b11d079758d28 Mon Sep 17 00:00:00 2001 From: ljcornel Date: Thu, 25 Apr 2024 14:00:53 +0200 Subject: [PATCH] Add `model_storage_id` to models when fetching model group --- geti_sdk/rest_clients/model_client.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/geti_sdk/rest_clients/model_client.py b/geti_sdk/rest_clients/model_client.py index 4bcc2b61..594f723f 100644 --- a/geti_sdk/rest_clients/model_client.py +++ b/geti_sdk/rest_clients/model_client.py @@ -77,6 +77,9 @@ def get_all_model_groups(self) -> List[ModelGroup]: group.algorithm = self.supported_algos.get_by_model_template( model_template_id=group.model_template_id ) + for model in group.models: + # set the model storage id, to link models to their parent group + model.model_storage_id = group.id return model_groups def get_latest_model_for_all_model_groups(self) -> List[Model]: