From c77d17e5e20fbf11d62c3ce8915cde33117b3447 Mon Sep 17 00:00:00 2001 From: Marko Djordjevic Date: Sat, 21 Feb 2026 09:54:51 +0100 Subject: [PATCH] Fix MLflow psycopg2 missing by installing psycopg2-binary at startup The official MLflow image lacks psycopg2. Install it via pip before starting the server. Also fix artifact root to use absolute path inside the mounted volume. Co-Authored-By: Claude Sonnet 4.6 --- docker-compose.yml | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index c46f0dd..29dcce9 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -68,12 +68,15 @@ services: - "127.0.0.1:5000:5000" volumes: - mlflow-data:/mlflow - command: > - mlflow server - --backend-store-uri "postgresql://${POSTGRES_USER}:${POSTGRES_PASSWORD}@postgres:5432/${POSTGRES_DB}" - --default-artifact-root ./mlruns/artifacts - --host 0.0.0.0 - --port 5000 + entrypoint: ["/bin/sh", "-c"] + command: + - | + pip install --quiet psycopg2-binary && + mlflow server + --backend-store-uri "postgresql://${POSTGRES_USER}:${POSTGRES_PASSWORD}@postgres:5432/${POSTGRES_DB}" + --default-artifact-root /mlflow/artifacts + --host 0.0.0.0 + --port 5000 restart: unless-stopped postgres: