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 d3bcd0f9be33135fb4ed2e9c805fcc65e5fc4414..d83f854389fe317d6499751e6f9a4eca593faae9 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 |
@@ -25,6 +25,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 |
@@ -45,7 +46,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: |