Index: tools/perf/profile_creators/profile_generator.py |
diff --git a/tools/perf/profile_creators/profile_generator.py b/tools/perf/profile_creators/profile_generator.py |
index dc683828d711e0fc4357e0ec0ce8551881012a91..4c52665525fd13448a736f337bfb058358307e53 100644 |
--- a/tools/perf/profile_creators/profile_generator.py |
+++ b/tools/perf/profile_creators/profile_generator.py |
@@ -13,11 +13,11 @@ |
import tempfile |
from profile_creators import profile_extender |
+from telemetry.core import discover |
from telemetry.core import util |
from telemetry.internal.browser import browser_finder |
from telemetry.internal.browser import browser_options |
from telemetry.internal import story_runner |
-from telemetry.util import classes_util |
def _DiscoverProfileExtenderClasses(): |
@@ -25,7 +25,7 @@ |
os.pardir, 'perf', 'profile_creators')) |
base_dir = os.path.abspath(os.path.join(profile_extenders_dir, os.pardir)) |
- profile_extenders_unfiltered = classes_util.DiscoverClassesByClassName( |
+ profile_extenders_unfiltered = discover.DiscoverClasses( |
profile_extenders_dir, base_dir, profile_extender.ProfileExtender) |
# Remove 'extender' suffix from keys. |