From c70d221a3094e7122fc8ea1bc59f2262dc557413 Mon Sep 17 00:00:00 2001
From: Guillaume Schaer <gschaer@users.noreply.github.com>
Date: Fri, 14 Feb 2025 12:23:03 +0100
Subject: [PATCH] Fix undefined PROJECT_NAME variable, remove deprecated
 compose version, and update restart policy

- Declare ARG PROJECT_NAME in the Dockerfile to resolve an undefined variable warning.
- Remove deprecated 'version: "3"' from docker-compose.yml.
- Update the container restart policy from "on-failure:5" to "unless-stopped" for more robust behavior.
---
 Dockerfile         | 1 +
 docker-compose.yml | 5 +++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 879c0fe..e0747cb 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,6 @@
 FROM mageai/mageai:latest
 
+ARG PROJECT_NAME
 ARG USER_CODE_PATH=/home/src/${PROJECT_NAME}
 
 # Note: this overwrites the requirements.txt file in your new project on first run. 
diff --git a/docker-compose.yml b/docker-compose.yml
index 832c208..a3135c5 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,4 +1,3 @@
-version: '3'
 services:
   magic:
     image: mageai/mageai:latest
@@ -8,6 +7,8 @@ services:
     build:
       context: .
       dockerfile: Dockerfile
+      args:
+        PROJECT_NAME: ${PROJECT_NAME}
     environment:
       USER_CODE_PATH: /home/src/${PROJECT_NAME}
       ENV: ${ENV}
@@ -15,4 +16,4 @@ services:
       - 6789:6789
     volumes:
       - .:/home/src/
-    restart: on-failure:5
+    restart: unless-stopped