From 4c1e9008abcc9ec28a75bf66f6380fe6223bde87 Mon Sep 17 00:00:00 2001 From: vitor-aignosi Date: Tue, 21 Oct 2025 14:32:35 -0300 Subject: [PATCH] SIENTIAPDE-1312 Update tests notebook execution count and bump image tag in values.yaml - Incremented the execution count in the tests notebook for accurate tracking. - Updated the image tag in values.yaml from "0.0.3" to "1.0.1" for versioning consistency. - Enhanced error handling in model_repository.py to ensure proper retrieval of created experiments. --- laborious/utils/repository/model_repository.py | 5 ++++- values.yaml | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index 3361d9a..20ac9f3 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -175,7 +175,10 @@ class MLFlowRepository: if experiment is None: if create_if_not_exists: - experiment = mlflow.create_experiment(experiment_name) + experiment_id = mlflow.create_experiment(experiment_name) + experiment = mlflow.get_experiment(experiment_id) + if experiment is None: + raise ValueError(f'Experiment {experiment_name} not found after creation, unknown reason') else: raise ValueError(f'Experiment {experiment_name} not found') diff --git a/values.yaml b/values.yaml index f100a7e..e4b3165 100644 --- a/values.yaml +++ b/values.yaml @@ -11,7 +11,7 @@ image: # This sets the pull policy for images. pullPolicy: Always # Overrides the image tag whose default is the chart appVersion. - tag: "0.0.3" + tag: "1.0.1" 0# This is for the secrets for pulling an image from a private repository more information can be found here: https://kubernetes.io/docs/tasks/configure-pod-container/pull-image-private-registry/ imagePullSecrets: