Index: tools/telemetry/telemetry/page/page_test_runner.py |
diff --git a/tools/telemetry/telemetry/page/page_test_runner.py b/tools/telemetry/telemetry/page/page_test_runner.py |
index 33ca9552fe0fe6b25e12c10d4433c18eea6f2a7a..3d75ef891cc69852d86776a9cc62dbfdd4e97f5f 100644 |
--- a/tools/telemetry/telemetry/page/page_test_runner.py |
+++ b/tools/telemetry/telemetry/page/page_test_runner.py |
@@ -108,8 +108,8 @@ class PageTestRunner(object): |
'pages': [{'url': page_set_arg}] |
}, os.path.dirname(__file__)) |
- # We've been given a page set JSON. Load it. |
- if page_set_arg.endswith('.json'): |
+ # We've been given a page set JSON or PYTHON. Load it. |
+ if page_set_arg.endswith('.json') or page_set_arg.endswith('.py'): |
return page_set.PageSet.FromFile(page_set_arg) |
# We've been given a file or directory. Create a page set containing it. |