Merge pull request #133 from microsoft/fix/task-fix

add task_name->expert_name for multi-task
microsoft · Nov 10, 2024 · 8307edc · 8307edc
2 parents bcffc15 + a59a992
commit 8307edc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mttl/models/containers/selectors/base.py b/mttl/models/containers/selectors/base.py
@@ -349,7 +349,7 @@ def add_expert(
                     logger.warning(
                         f"Task name {task_name} already assigned to expert {self._task_to_expert_name[task_name]}"
                     )
-            self._task_to_expert_name[task_name] = expert_name
+                self._task_to_expert_name[task_name] = expert_name
 
         # standard bookkeeping for all selectors
         if is_default: