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: