From d7f583701cbe75df2a40d2cf587139e48dca2b48 Mon Sep 17 00:00:00 2001
From: Marko Djordjevic <marko.homoludens@gmail.com>
Date: Sat, 21 Feb 2026 09:35:15 +0100
Subject: [PATCH] change mlflow to use database instead of mlflow folder

---
 docker-compose.yml | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index f5b6a68..c46f0dd 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -48,6 +48,7 @@ services:
       - POSTGRES_USER=${POSTGRES_USER}
       - POSTGRES_PASSWORD=${POSTGRES_PASSWORD}
       - POSTGRES_DB=${POSTGRES_DB}
+      - POSTGRES_HOST=postgres
     restart: unless-stopped
     depends_on:
       postgres:
@@ -69,10 +70,10 @@ services:
       - mlflow-data:/mlflow
     command: >
       mlflow server
-      --backend-store-uri /mlflow
-      --default-artifact-root /mlflow/artifacts
-      --host 0.0.0.0
-      --port 5000
+        --backend-store-uri "postgresql://${POSTGRES_USER}:${POSTGRES_PASSWORD}@postgres:5432/${POSTGRES_DB}"
+        --default-artifact-root ./mlruns/artifacts
+        --host 0.0.0.0
+        --port 5000
     restart: unless-stopped
 
   postgres: