Index: build/android/pylib/local/device/local_device_instrumentation_test_run.py |
diff --git a/build/android/pylib/local/device/local_device_instrumentation_test_run.py b/build/android/pylib/local/device/local_device_instrumentation_test_run.py |
index ee6f7a45c2b8e4fde7419885ea3d7778e9737365..26d10f4d1d7ed797a4682a0acae6ff13b382281e 100644 |
--- a/build/android/pylib/local/device/local_device_instrumentation_test_run.py |
+++ b/build/android/pylib/local/device/local_device_instrumentation_test_run.py |
@@ -13,6 +13,7 @@ from devil.android import flag_changer |
from devil.utils import reraiser_thread |
from pylib import valgrind_tools |
from pylib.base import base_test_result |
+from pylib.local.device import local_device_environment |
from pylib.local.device import local_device_test_run |
@@ -67,7 +68,7 @@ class LocalDeviceInstrumentationTestRun( |
else: |
return d |
- @local_device_test_run.handle_shard_failures_with( |
+ @local_device_environment.handle_shard_failures_with( |
self._env.BlacklistDevice) |
def individual_device_set_up(dev, host_device_tuples): |
def install_apk(): |
@@ -148,7 +149,7 @@ class LocalDeviceInstrumentationTestRun( |
self._test_instance.GetDataDependencies()) |
def TearDown(self): |
- @local_device_test_run.handle_shard_failures_with( |
+ @local_device_environment.handle_shard_failures_with( |
self._env.BlacklistDevice) |
def individual_device_tear_down(dev): |
if str(dev) in self._flag_changers: |