Index: tools/telemetry/telemetry/test/discover.py |
diff --git a/tools/telemetry/telemetry/discover.py b/tools/telemetry/telemetry/test/discover.py |
similarity index 92% |
rename from tools/telemetry/telemetry/discover.py |
rename to tools/telemetry/telemetry/test/discover.py |
index a2a85dd2776ee9eefd5ae60b90af47cf3807af6b..f2398fb8b3e761ed04bb276367de1016fa182c63 100644 |
--- a/tools/telemetry/telemetry/discover.py |
+++ b/tools/telemetry/telemetry/test/discover.py |
@@ -6,7 +6,8 @@ import logging |
import os |
import traceback |
-def Discover(start_dir, suffix, clazz, import_error_should_raise=False): |
+def Discover(start_dir, top_level_dir, suffix, clazz, |
+ import_error_should_raise=False): |
"""Discover all classes in |start_dir| which subclass |clazz|. |
Args: |
@@ -19,7 +20,6 @@ def Discover(start_dir, suffix, clazz, import_error_should_raise=False): |
Returns: |
dict of {module_name: class}. |
""" |
- top_level_dir = os.path.join(start_dir, '..') |
classes = {} |
for dirpath, _, filenames in os.walk(start_dir): |
for filename in filenames: |