From d558c66d3c1d71a372f2634314f8f1de6785b989 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Mon, 10 Feb 2020 10:47:13 +0200
Subject: [PATCH] Do not stop experiments if network is down

---
 trains_agent/commands/worker.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/trains_agent/commands/worker.py b/trains_agent/commands/worker.py
index 3a5da78..30e80ea 100644
--- a/trains_agent/commands/worker.py
+++ b/trains_agent/commands/worker.py
@@ -229,6 +229,8 @@ class TaskStopSignal(object):
             return self._test()
         except Exception as ex:
             self.command.log_traceback(ex)
+            # make sure we break nothing
+            return TaskStopSignal.default
 
     def _test(self):
         # type: () -> TaskStopReason
@@ -780,7 +782,7 @@ class Worker(ServiceCommandSection):
                     if daemon:
                         self.send_logs(
                             task_id=task_id,
-                            lines=["User aborted: stopping task\n"],
+                            lines=["User aborted: stopping task ({})\n".format(str(stop_reason))],
                             level="ERROR",
                         )
                         kill_all_child_processes(process.pid)