Index: tools/perf/PRESUBMIT.py |
diff --git a/tools/perf/PRESUBMIT.py b/tools/perf/PRESUBMIT.py |
index 1b7d74f8ddb6774fa58ddde29c7aed1b5385f299..c606204c6405cd223318dec88a632dc0b786df94 100644 |
--- a/tools/perf/PRESUBMIT.py |
+++ b/tools/perf/PRESUBMIT.py |
@@ -39,8 +39,10 @@ def _CheckWprShaFiles(input_api, output_api): |
"""Check whether the wpr sha files have matching URLs.""" |
old_sys_path = sys.path |
try: |
- # TODO: The cloud_storage module is in telemetry. |
- sys.path = [os.path.join('..', 'telemetry')] + sys.path |
+ perf_dir = input_api.PresubmitLocalPath() |
+ catapult_path = os.path.abspath(os.path.join( |
+ perf_dir, '..', '..', 'third_party', 'catapult', 'catapult_base')) |
+ sys.path.insert(1, catapult_path) |
from catapult_base import cloud_storage |
finally: |
sys.path = old_sys_path |