fixed mlflow docker

Author: floschne

.gitignore

@@ -9,6 +9,7 @@ docker/weaviate_data/
 docker/weaviate_data-dev/
 docker/cache/
 docker/mlflow_runs/
+docker/mlflow_artifacts/
 
 config/config.dev.yaml
 config/config.prod.yaml

docker/compose.yaml

@@ -21,8 +21,8 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              device_ids: [ "1" ]
-              capabilities: [ gpu ]
+              device_ids: ["1"]
+              capabilities: [gpu]
 
   weaviate:
     image: cr.weaviate.io/semitechnologies/weaviate:1.26.5
@@ -49,7 +49,6 @@ services:
     profiles:
       - weaviate
 
-
   weaviate-dev:
     image: cr.weaviate.io/semitechnologies/weaviate:1.26.5
     user: ${UID:-1000}:${GID:-1000}
@@ -80,7 +79,7 @@ services:
 
   mlflow:
     image: ghcr.io/mlflow/mlflow:latest
-    user: 1000:1000
+    user: ${UID:-1000}:${GID:-1000}
     command:
       - mlflow
       - ui
@@ -91,6 +90,7 @@ services:
     ports:
       - "${MLFLOW_EXPOSED:-5000}:5000"
     volumes:
+      - ./mlflow_artifacts:/mlartifacts
       - ./mlflow_runs:/mlruns
     restart: on-failure
     networks:
@@ -103,7 +103,7 @@ services:
     ports:
       - "${FRONTEND_EXPOSED:-8080}:80"
     healthcheck:
-      test: [ "CMD-SHELL", "curl -f http://localhost:80/heartbeat || exit 1" ]
+      test: ["CMD-SHELL", "curl -f http://localhost:80/heartbeat || exit 1"]
       interval: 30s
       timeout: 10s
       retries: 5
@@ -115,7 +115,6 @@ services:
     profiles:
       - frontend
 
-
   fundus-murag-backend:
     image: p0w3r/fundus_murag_backend:latest
     user: ${UID:-1000}:${GID:-1000}

docker/setup-folders.sh

@@ -42,3 +42,9 @@ if [ ! -d "mlflow_runs" ]; then
 else
   echo "mlflow_runs already exists"
 fi
+
+if [ ! -d "mlflow_artifacts" ]; then
+  mkdir -v mlflow_artifacts
+else
+  echo "mlflow_artifacts already exists"
+fi