From 79d0abe707341917aff1da338eacb8b1c2fb66bf Mon Sep 17 00:00:00 2001
From: Valentin Schabschneider <valentin.schabschneider@gmail.com>
Date: Fri, 26 Jul 2024 18:09:54 +0200
Subject: [PATCH] Add NO_DOCKER flag to clearml-agent-services entrypoint
 (#206)

---
 docker/services/entrypoint.sh | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/docker/services/entrypoint.sh b/docker/services/entrypoint.sh
index 42fd2ff..3df488d 100755
--- a/docker/services/entrypoint.sh
+++ b/docker/services/entrypoint.sh
@@ -33,4 +33,10 @@ if [ -z "$CLEARML_AGENT_NO_UPDATE" ]; then
   fi
 fi
 
-clearml-agent daemon $DAEMON_OPTIONS --queue $QUEUES --docker "${CLEARML_AGENT_DEFAULT_BASE_DOCKER:-$TRAINS_AGENT_DEFAULT_BASE_DOCKER}" --cpu-only ${CLEARML_AGENT_EXTRA_ARGS:-$TRAINS_AGENT_EXTRA_ARGS}
+DOCKER_ARGS="--docker \"${CLEARML_AGENT_DEFAULT_BASE_DOCKER:-$TRAINS_AGENT_DEFAULT_BASE_DOCKER}\""
+
+if [ -n "$CLEARML_AGENT_NO_DOCKER" ]; then
+  DOCKER_ARGS=""
+fi
+
+clearml-agent daemon $DAEMON_OPTIONS --queue $QUEUES $DOCKER_ARGS --cpu-only ${CLEARML_AGENT_EXTRA_ARGS:-$TRAINS_AGENT_EXTRA_ARGS}