| Index: build/android/pylib/local/device/local_device_test_run.py
|
| diff --git a/build/android/pylib/local/device/local_device_test_run.py b/build/android/pylib/local/device/local_device_test_run.py
|
| index ba3b9783d0603ffc7d691ff7e55ce67e695bfe24..481bed427baddbfc496d0c07cf1b3c81bcb6650b 100644
|
| --- a/build/android/pylib/local/device/local_device_test_run.py
|
| +++ b/build/android/pylib/local/device/local_device_test_run.py
|
| @@ -2,6 +2,7 @@
|
| # Use of this source code is governed by a BSD-style license that can be
|
| # found in the LICENSE file.
|
|
|
| +import functools
|
| import logging
|
|
|
| from devil.android import device_errors
|
| @@ -18,6 +19,7 @@ def handle_shard_failures(f):
|
| f: the function being decorated. The function must take at least one
|
| argument, and that argument must be the device.
|
| """
|
| + @functools.wraps(f)
|
| def wrapper(dev, *args, **kwargs):
|
| try:
|
| return f(dev, *args, **kwargs)
|
| @@ -29,7 +31,6 @@ def handle_shard_failures(f):
|
| logging.exception('Shard died: %s(%s)', f.__name__, str(dev))
|
| return None
|
|
|
| - wrapper.__name__ = f.__name__
|
| return wrapper
|
|
|
|
|
|
|