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 200ee4e41676d4519a1889da187bccbb90c6c0cc..2a93c68144cb7890b6440fd91a66d1b62c03808a 100644 |
--- a/build/android/pylib/local/device/local_device_test_run.py |
+++ b/build/android/pylib/local/device/local_device_test_run.py |
@@ -31,7 +31,7 @@ def handle_shard_failures_with(on_failure): |
Args: |
f: the function being decorated. The function must take at least one |
jbudorick
2015/10/28 23:34:05
nit: update this now that the function must take a
bpastene
2015/10/28 23:41:57
The decorated function doesn't change. The on_fail
jbudorick
2015/10/28 23:45:15
oh, right. derp.
|
argument, and that argument must be the device. |
- on_failure: A unary function to call on failure. |
+ on_failure: A binary function to call on failure. |
""" |
def decorator(f): |
@functools.wraps(f) |
@@ -45,7 +45,7 @@ def handle_shard_failures_with(on_failure): |
except device_errors.DeviceUnreachableError: |
logging.exception('Shard died: %s(%s)', f.__name__, str(dev)) |
if on_failure: |
- on_failure(dev) |
+ on_failure(dev, f.__name__) |
return None |
return wrapper |