fix

microsoft · Aug 1, 2024 · 265678c · 265678c
1 parent f74c6a5
commit 265678c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/mttl/models/containers/__init__.py b/mttl/models/containers/__init__.py
@@ -89,8 +89,8 @@ def create_selector_for_container(
     transformer,
     container,
     modifier_name: str,
-    selector_config: SelectorConfig,
-    selector_cache: SelectorsCache,
+    selector_config: SelectorConfig = None,
+    selector_cache: SelectorsCache = None,
 ) -> Selector:
     if container.selector is not None and container.selector.config == selector_config:
         # selector already exists and has the same config
@@ -130,8 +130,8 @@ def create_selector_for_container(
 def replace_selector_for_container(
     transformer,
     modifier_name: str,
-    selector_config: SelectorConfig,
-    selector_cache: SelectorsCache,
+    selector_config: SelectorConfig = None,
+    selector_cache: SelectorsCache = None,
     force_replace: bool = False,
 ) -> Tuple[int, int]:
     """

diff --git a/mttl/models/expert_model.py b/mttl/models/expert_model.py
@@ -505,7 +505,6 @@ def set_selector(
     ):
         from mttl.models.containers import replace_selector_for_container
 
-        selector_config = self._get_selector_config(modifier_name)
         n_selectors, n_selectors_views = replace_selector_for_container(
             self.model,
             modifier_name,