Support gist for opening issue on mlflow/mlflow

.env

MLFLOW_DB_NAME=mlflow_db
MLFLOW_DB_PASSWORD=use_vault
MLFLOW_DB_CONNECTION_URI=postgresql+asyncpg://postgres:${MLFLOW_DB_PASSWORD}@mlflow-db:5432/${MLFLOW_DB_NAME}

docker-compose.yml

version: '3.8'

volumes:
  mlflow_artifact_vol:
  mlflow_db_vol:
    external: true


networks:
  mlflow:

services:

  mlflow:
    # restart: always
    build:
      context: ./mlflow
    command: [
      "--host", "0.0.0.0",
      "-p", "5000",
      "--default-artifact-root", "/home/mlflow",
      "--backend-store-uri", "${MLFLOW_DB_CONNECTION_URI}"
    ]
    volumes:
      - mlflow_artifact_vol:/home/mlflow
    depends_on: [mlflow-db]
    networks:
      - mlflow
    ports:
      - "5000:5000"  # maps to the server / host (i.e. your machine)

  mlflow-db:
    # restart: always
    image: postgres:14
    volumes:
      - mlflow_db_vol:/var/lib/postgresql/data
    environment:
      - POSTGRES_PASSWORD=${MLFLOW_DB_PASSWORD}
      - POSTGRES_DB=${MLFLOW_DB_NAME}
    networks:
      - mlflow
    expose:
      - 5432

Dockerfile

FROM python:slim-buster

RUN /usr/local/bin/python -m pip install --upgrade pip
RUN pip install mlflow==1.21
RUN pip install psycopg2-binary asyncpg
RUN mkdir -p /home/mlflow

EXPOSE 5000
ENTRYPOINT [ "mlflow", "server" ]
CMD [ "--host", "0.0.0.0" ]