From 909f5cea2dc0e335d80f862561bb8dafbd5716a1 Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Tue, 17 Nov 2020 23:08:25 +0200 Subject: [PATCH] Fix running remotely with no configuration should not crash but output a warning (issue #243) --- trains/task.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/trains/task.py b/trains/task.py index 84371915..8ba692b3 100644 --- a/trains/task.py +++ b/trains/task.py @@ -1056,9 +1056,21 @@ class Task(_Task): self._set_model_config(config_dict=configuration) configuration = ProxyDictPostWrite(self, _update_config_dict, **configuration) else: + # noinspection PyBroadException + try: + remote_configuration = self._get_configuration_dict(name=name) \ + if multi_config_support else self._get_model_config_dict() + except Exception: + remote_configuration = None + + if remote_configuration is None: + LoggerRoot.get_base_logger().warning( + "Could not retrieve remote configuration named \'{}\'\n" + "Using default configuration: {}".format(name, str(configuration))) + return configuration + configuration.clear() - configuration.update(self._get_configuration_dict(name=name) if multi_config_support - else self._get_model_config_dict()) + configuration.update(remote_configuration) configuration = ProxyDictPreWrite(False, False, **configuration) return configuration