From a345c328203217fb337da0771ff5e2d8391273a3 Mon Sep 17 00:00:00 2001 From: Alessandro Sordoni Date: Sun, 10 Nov 2024 12:37:08 -0800 Subject: [PATCH] any not all --- mttl/models/containers/selectors/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mttl/models/containers/selectors/base.py b/mttl/models/containers/selectors/base.py index 507189e60..c9433cc6c 100644 --- 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 )