| Index: build/android/pylib/device/decorators.py
|
| diff --git a/build/android/pylib/device/decorators.py b/build/android/pylib/device/decorators.py
|
| index 074a0fcac14131abe15fc3cdbf719c15849f7908..73c13da8b09b9c0c2ffe6053f6fac529801ed997 100644
|
| --- a/build/android/pylib/device/decorators.py
|
| +++ b/build/android/pylib/device/decorators.py
|
| @@ -11,6 +11,7 @@ import os
|
| import sys
|
| import threading
|
|
|
| +from pylib import cmd_helper
|
| from pylib import constants
|
| from pylib.device import device_errors
|
| from pylib.utils import reraiser_thread
|
| @@ -56,13 +57,16 @@ def _TimeoutRetryWrapper(f, timeout_func, retries_func, pass_values=False):
|
| return timeout_retry.Run(impl, timeout, retries)
|
| except old_errors.WaitForResponseTimedOutError as e:
|
| raise device_errors.CommandTimeoutError(str(e)), None, (
|
| - sys.exc_info()[2])
|
| + sys.exc_info()[2])
|
| except old_errors.DeviceUnresponsiveError as e:
|
| raise device_errors.DeviceUnreachableError(str(e)), None, (
|
| - sys.exc_info()[2])
|
| + sys.exc_info()[2])
|
| except reraiser_thread.TimeoutError as e:
|
| raise device_errors.CommandTimeoutError(str(e)), None, (
|
| - sys.exc_info()[2])
|
| + sys.exc_info()[2])
|
| + except cmd_helper.TimeoutError as e:
|
| + raise device_errors.CommandTimeoutError(str(e)), None, (
|
| + sys.exc_info()[2])
|
| return TimeoutRetryWrapper
|
|
|
|
|
|
|