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: