Merge pull request #24 from ovaleanu/mlflow-tracking-addon

chore: MLflow version update

helm-charts/mlflow-tracking/Chart.yaml

@@ -3,5 +3,5 @@ name: mlflow-tracking-server
 icon: https://mlflow.org/images/MLflow-logo-final-white-TM.png
 description: A Helm chart for MLflow Tracking
 type: application
-version: 0.1.0
-appVersion: "2.7.1"
+version: 0.2.0
+appVersion: "2.11.1"

helm-charts/mlflow-tracking/Dockerfile

@@ -3,7 +3,7 @@
 # This docker image is available at public.ecr.aws/data-on-eks/mlflow
 
 
-FROM python:3.11-slim
+FROM python:3.12-slim
 
 # Build dependencies
 RUN apt-get update -y && \
@@ -12,7 +12,7 @@ RUN apt-get update -y && \
 #
 
 RUN python -m pip install --upgrade pip && \
-  pip install --no-cache-dir mlflow[extras]==2.7.1 boto3 psycopg2-binary PyMySQL
+  pip install --no-cache-dir mlflow[extras]==2.11.1 boto3 psycopg2-binary PyMySQL
 
 EXPOSE 5000
 

helm-charts/mlflow-tracking/values.yaml

@@ -5,7 +5,7 @@
 image:
   repository: public.ecr.aws/data-on-eks/mlflow
   pullPolicy: Always
-  tag: 2.7.1
+  tag: 2.11.1
 
 imagePullSecrets: []
 

mlflow-tracking.tf

@@ -4,7 +4,7 @@ resource "helm_release" "mlflow_tracking" {
   name                       = try(var.mlflow_tracking_helm_config["name"], "mlflow-tracking")
   repository                 = try(var.mlflow_tracking_helm_config["repository"], null)
   chart                      = try(var.mlflow_tracking_helm_config["chart"], "${path.module}/helm-charts/mlflow-tracking")
-  version                    = try(var.mlflow_tracking_helm_config["version"], "0.1.0")
+  version                    = try(var.mlflow_tracking_helm_config["version"], "0.2.0")
   timeout                    = try(var.mlflow_tracking_helm_config["timeout"], 300)
   values                     = try(var.mlflow_tracking_helm_config["values"], null)
   create_namespace           = try(var.mlflow_tracking_helm_config["create_namespace"], true)