From 86da29b5601f98eb06aaa66c03ff2e71b20863fa Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Tue, 29 Sep 2020 19:18:49 +0300 Subject: [PATCH] Fix TrainsJob hyperparams overriding order was not guaranteed --- trains/automation/job.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/trains/automation/job.py b/trains/automation/job.py index 657a9f60..cc2d9a92 100644 --- a/trains/automation/job.py +++ b/trains/automation/job.py @@ -37,9 +37,9 @@ class TrainsJob(object): self.task.set_tags(list(set(self.task.get_tags()) | set(tags))) self.task_parameter_override = None if parameter_override: - params = self.task.get_parameters_as_dict() + params = self.task.get_parameters(backwards_compatibility=False) params.update(parameter_override) - self.task.set_parameters_as_dict(params) + self.task.set_parameters(params) self.task_parameter_override = dict(**parameter_override) if task_overrides: # todo: make sure it works