diff --git a/golem/core/optimisers/adaptive/mab_agents/mab_agent.py b/golem/core/optimisers/adaptive/mab_agents/mab_agent.py index 538c2ed2..91b0274d 100644 --- a/golem/core/optimisers/adaptive/mab_agents/mab_agent.py +++ b/golem/core/optimisers/adaptive/mab_agents/mab_agent.py @@ -82,7 +82,7 @@ def save(self, path_to_save: Optional[str] = None): if not path_to_save.endswith('.pkl'): os.makedirs(path_to_save, exist_ok=True) mabs_num = [int(name.split('_')[0]) for name in os.listdir(path_to_save) - if re.fullmatch(r'\d_mab.pkl', name)] + if re.fullmatch(r'\d*_mab.pkl', name)] if not mabs_num: max_saved_mab = 0 else: