any not all

microsoft · Nov 10, 2024 · a345c32 · a345c32
1 parent 191358c
commit a345c32
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
@@ -513,7 +513,7 @@ def forward(self, input, **kwargs) -> BatchExpertsSelectorOutput:
             task_names = self.routing_infos.task_names
 
             if (
-                any(
+                all(
                     task_name not in self.task_to_expert_name
                     for task_name in task_names
                 )