add retry (#38)

aistairc · Aug 19, 2022 · 8f6df87 · 8f6df87
1 parent 5766730
commit 8f6df87
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/aiaccel/master/abci_master.py b/aiaccel/master/abci_master.py
@@ -4,6 +4,7 @@
 import aiaccel
 import subprocess
 import time
+from aiaccel.util.filesystem import retry
 
 
 class AbciMaster(AbstractMaster):
@@ -40,6 +41,7 @@ def pre_process(self) -> None:
             "run_*.sh"
         )
 
+    @retry(_MAX_NUM=60, _DELAY=1.0)
     def get_stats(self) -> None:
         """Get a current status and update.