feat(mlflow): Bump up to v1.12.1

StatCan · Dec 27, 2020 · b901376 · b901376
1 parent e340efc
commit b901376
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/mlflow/Dockerfile b/mlflow/Dockerfile
@@ -7,18 +7,18 @@ RUN apt-get update && \
     rm -rf /var/lib/apt/lists/*
 
 RUN pip --no-cache-dir install --quiet --no-use-pep517 \
-      'mlflow==1.9.1' \
-      'azure-storage-blob==12.3.2' \
-      'azure-storage-file-share==12.1.2' \
-      'azure-storage-file-datalake==12.0.2' \
-      'azure-storage-queue==12.1.2' \
-      'msrestazure~=0.6.3' \
-      'psycopg2-binary==2.8.5'
+      'mlflow==1.12.1' \
+      'azure-storage-blob==12.6.0' \
+      'azure-storage-file-share==12.3.0' \
+      'azure-storage-file-datalake==12.2.0' \
+      'azure-storage-queue==12.1.4' \
+      'msrestazure~=0.6.4' \
+      'psycopg2-binary==2.8.6'
 
 # REQUIRED FOR AZURE:
-# ENV AZURE_STORAGE_ACCESS_KEY DefaultEndpointsProtocol=https;EndpointSuffix=core.windows.net;AccountName=XXXXX;AccountKey=XXXXX
-# ENV MLFLOW_SERVER_FILE_STORE /mnt/mlruns
-# ENV MLFLOW_SERVER_DEFAULT_ARTIFACT_ROOT wasbs://mlflow@XXXXX.blob.core.windows.net/mlartifacts
+# ENV AZURE_STORAGE_CONNECTION_STRING "XXXXX"
+# ENV MLFLOW_SERVER_FILE_STORE "sqlite:///mnt/mlruns/mlflow.db"
+# ENV MLFLOW_SERVER_DEFAULT_ARTIFACT_ROOT "wasbs://XXXXX@XXXXX.blob.core.windows.net/mlartifacts"
 
 ENV MLFLOW_SERVER_HOST 0.0.0.0
 ENV MLFLOW_SERVER_PORT 5000

diff --git a/mlflow/entrypoint.sh b/mlflow/entrypoint.sh
@@ -3,6 +3,8 @@
 export MLFLOW_SERVER_FILE_STORE="${MLFLOW_SERVER_FILE_STORE:-./mlruns}"
 export MLFLOW_SERVER_DEFAULT_ARTIFACT_ROOT="${MLFLOW_SERVER_DEFAULT_ARTIFACT_ROOT:-./mlruns}"
 
+mlflow db upgrade $MLFLOW_SERVER_FILE_STORE
+
 mlflow server \
     --backend-store-uri "$MLFLOW_SERVER_FILE_STORE" \
     --default-artifact-root "$MLFLOW_SERVER_DEFAULT_ARTIFACT_ROOT" \