add fallback for non-integer delay

microsoft · Jan 10, 2024 · bc79e0b · bc79e0b
1 parent bb49590
commit bc79e0b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/raiutils/raiutils/common/retries.py b/raiutils/raiutils/common/retries.py
@@ -33,8 +33,11 @@ def retry_function(function, action_name, err_msg,
             print(e)
             if i + 1 != max_retries:
                 print("Will retry after {0} seconds".format(retry_delay))
-                for _ in range(retry_delay):
-                    time.sleep(1)
+                try:
+                    for _ in range(retry_delay):
+                        time.sleep(1)
+                except TypeError:
+                    time.sleep(retry_delay)
                 retry_delay = retry_delay * 2
     else:
         raise RuntimeError(err_msg)