diff --git a/mttl/models/containers/selectors/base.py b/mttl/models/containers/selectors/base.py index 0a6622b3..04420ea3 100644 --- a/mttl/models/containers/selectors/base.py +++ b/mttl/models/containers/selectors/base.py @@ -101,6 +101,9 @@ def __len__(self): def __getitem__(self, key): return self.selectors[key] + def get(self, key): + return self.selectors.get(key, default=TaskNameSelectorConfig()) + def keys(self): return self.selectors.keys()