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 |