Add mlflow db upgrade

DougTrajano · Nov 27, 2022 · b7224e7 · b7224e7
1 parent 8fce928
commit b7224e7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/mlflow.sh b/scripts/mlflow.sh
@@ -49,6 +49,9 @@ if [[ -z "${MLFLOW_BACKEND_URI}" ]]; then
     unset MLFLOW_DB_PORT
 fi
 
+echo "Upgrading database..."
+mlflow db upgrade ${MLFLOW_BACKEND_URI}
+
 echo "Starting mlflow server"
 
 exec mlflow server --host 0.0.0.0 --port 5001 \