Skip to content
4 changes: 2 additions & 2 deletions Dockerfile.mlflow
Original file line number Diff line number Diff line change
Expand Up @@ -3,5 +3,5 @@ FROM ghcr.io/mlflow/mlflow:v3.7.0
# The base image does not include various dependencies that are needed for
# the MLflow server. We assume a postgres backend, so we need psycopg2.
# We also need boto3 for S3 support, and google-cloud-storage for GCS support.
RUN pip install mlflow[auth]==3.7.0 psycopg2-binary==2.9.11 boto3==1.42.51 \
google-cloud-storage==3.9.0
RUN pip install mlflow[auth]==3.7.0 psycopg2-binary==2.9.12 boto3==1.43.21 \
google-cloud-storage==3.10.1
6 changes: 3 additions & 3 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,9 @@ dependencies = [
"modelbench[composer] @ git+https://github.com/mlcommons/modelbench.git",
# the below 4 are tied to the Dockerfile.mlflow versions; need to be upgraded together
"mlflow[auth]==3.7.0",
"psycopg2-binary==2.9.11",
"boto3==1.42.51",
"google-cloud-storage==3.9.0",
"psycopg2-binary==2.9.12",
"boto3==1.43.21",
"google-cloud-storage==3.10.1",
]

[project.scripts]
Expand Down
Loading
Loading