commit f24be08339541fdb24a0c79c0adccc61c21ad48e
author Fang Deng <fdeng@google.com> Sat Feb 10 10:09:55 2018 -0800
committer Fang Deng <fdeng@google.com> Fri Feb 16 01:50:38 2018 +0000
tree a3f4fbed0472336bd1d8f88a4e577e1bac18f473
parent 88bdc7d3bed40502c3082ce458b7b295c6333af5

Implement retry related functions

TEST: manual test
Change-Id: I165ab62bf7be96f629982676822543ac02940816

public/acloud_kernel/kernel_swapper.py

diff --git a/public/acloud_kernel/kernel_swapper.py b/public/acloud_kernel/kernel_swapper.py
index ca48c26..d30eb9e 100755
--- a/public/acloud_kernel/kernel_swapper.py
+++ b/public/acloud_kernel/kernel_swapper.py
@@ -147,8 +147,8 @@
             subprocess.CalledProcessError: For any non-zero return code of
                                            host_cmd.
         """
-        utils.GenericRetry(
-            handler=lambda e: isinstance(e, subprocess.CalledProcessError),
-            max_retry=2,
+        utils.Retry(
+            retry_checker=lambda e: isinstance(e, subprocess.CalledProcessError),
+            max_retries=2,
             functor=lambda cmd: subprocess.check_call(cmd, shell=True),
             cmd=host_cmd)