fix minor with saving

aimclub · Dec 9, 2023 · a6f2b90 · a6f2b90
1 parent 1fe1c0d
commit a6f2b90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: