Index: build/android/pylib/local/device/local_device_perf_test_run.py |
diff --git a/build/android/pylib/local/device/local_device_perf_test_run.py b/build/android/pylib/local/device/local_device_perf_test_run.py |
index 2981f49c5e1ed50853cff68c1834443e99f7bdf9..7beb822788142542cd5bcf8ce752f14d5c088110 100644 |
--- a/build/android/pylib/local/device/local_device_perf_test_run.py |
+++ b/build/android/pylib/local/device/local_device_perf_test_run.py |
@@ -24,6 +24,7 @@ from devil.utils import parallelizer |
from pylib import constants |
from pylib.base import base_test_result |
from pylib.constants import host_paths |
+from pylib.local.device import local_device_environment |
from pylib.local.device import local_device_test_run |
@@ -44,7 +45,7 @@ class TestShard(object): |
self._tests = tests |
self._timeout = timeout |
- @local_device_test_run.handle_shard_failures |
+ @local_device_environment.handle_shard_failures |
def RunTestsOnShard(self): |
results = base_test_result.TestRunResults() |
for test in self._tests: |