Fix HyperParameter optimizer to create Task of type optimizer

2025-03-03 18:52:12 +00:00 · 2020-08-23 01:06:59 +03:00 · 2020-08-23 01:06:59 +03:00 · 838c9cb0d2
commit 838c9cb0d2
parent e1bbc03001
1 changed files with 2 additions and 1 deletions
--- a/trains/automation/optimization.py
+++ b/trains/automation/optimization.py
@ -882,7 +882,8 @@ class HyperParameterOptimizer(object):
            self._task = Task.init(
                project_name=base_task.get_project_name(),
                task_name='Optimizing: {}'.format(base_task.name),
-            )  # TODO: add task_type=controller
+                task_type=Task.TaskTypes.optimizer,
+            )

        opts = dict(
            base_task_id=base_task_id,