From 9f77b3275fed217b73339749ef1eebf32a838a36 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Wed, 2 Feb 2022 15:34:11 +0200
Subject: [PATCH] Fix AutoScaler should not fail on any command, but only on
 the agent failing to launch

---
 clearml/automation/cloud_driver.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/clearml/automation/cloud_driver.py b/clearml/automation/cloud_driver.py
index 3dac773e..146e0537 100644
--- a/clearml/automation/cloud_driver.py
+++ b/clearml/automation/cloud_driver.py
@@ -14,7 +14,6 @@ bash_script_template = '''\
 #!/bin/bash
 
 set -x
-set -e
 
 apt-get update
 apt-get install -y \
@@ -43,6 +42,12 @@ source ~/.bashrc
 {bash_script}
 {driver_extra}
 python -m clearml_agent --config-file ~/clearml.conf daemon --queue '{queue}' {docker}
+
+if [[ $? -ne 0 ]]
+then
+  exit 1
+fi
+
 shutdown
 '''