From a6f2b903f4f57919ff137156588374b2e2285eca Mon Sep 17 00:00:00 2001 From: Pinchuk Maya Date: Sat, 9 Dec 2023 17:09:12 +0300 Subject: [PATCH] fix minor with saving --- golem/core/optimisers/adaptive/mab_agents/mab_agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: