From e7843c4dc41a525f3c1beedbb898b351b4f148fb Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Fri, 30 Oct 2020 10:01:42 +0200
Subject: [PATCH] Fix warning on optimization parameter is missing from base
 task id

---
 trains/automation/optimization.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/trains/automation/optimization.py b/trains/automation/optimization.py
index eaebff19..aaf6945d 100644
--- a/trains/automation/optimization.py
+++ b/trains/automation/optimization.py
@@ -572,7 +572,7 @@ class SearchStrategy(object):
         except ValueError:
             raise ValueError("Could not find base task id {}".format(self._base_task_id))
         # check if the hyper-parameters exist:
-        task_parameters = task.get_parameters_as_dict()
+        task_parameters = task.get_parameters(backwards_compatibility=False)
         missing_params = [h.name for h in self._hyper_parameters if h.name not in task_parameters]
         if missing_params:
             logger.warning('Could not find requested hyper-parameters {} on base task {}'.format(