From 0dc4d48d3e792862324cc55cf85f6680f28fc178 Mon Sep 17 00:00:00 2001
From: Mathis <matus@namesny.com>
Date: Wed, 3 Apr 2024 19:33:20 +0000
Subject: [PATCH] Add mlflow db upgrade init container

---
 apps/mlflow/mlflow-deployment.yaml | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/apps/mlflow/mlflow-deployment.yaml b/apps/mlflow/mlflow-deployment.yaml
index 576a28c..368c5e2 100644
--- a/apps/mlflow/mlflow-deployment.yaml
+++ b/apps/mlflow/mlflow-deployment.yaml
@@ -30,6 +30,15 @@ spec:
             name: minio-admin-secret
         - secretRef:
             name: minio-user-secret
+      - name: init-db-upgrade
+        image: git.namesny.com/cluster/mlflow:latest
+        envFrom:
+        - secretRef:
+            name: mlflow-secret
+        command: ["/bin/sh", "-c"]
+        args:
+          - mlflow db upgrade $MLFLOW_BACKEND_STORE_URI;
+            exit 0;
       containers:
       - name: mlflow
         image: git.namesny.com/cluster/mlflow:latest