Index: tools/telemetry/telemetry/user_story/user_story_set.py |
diff --git a/tools/telemetry/telemetry/user_story/user_story_set.py b/tools/telemetry/telemetry/user_story/user_story_set.py |
index a6771fd6adf762e163f7824b369c7310d9b19601..d2251bba0b1df3cbca68e07b81aee226b2bbfef0 100644 |
--- a/tools/telemetry/telemetry/user_story/user_story_set.py |
+++ b/tools/telemetry/telemetry/user_story/user_story_set.py |
@@ -6,7 +6,7 @@ import inspect |
import os |
from telemetry import user_story as user_story_module |
-from telemetry.page import page_set_archive_info |
+from telemetry.wpr import archive_info |
class UserStorySet(object): |
@@ -30,7 +30,7 @@ class UserStorySet(object): |
self.user_stories = [] |
self._archive_data_file = archive_data_file |
self._wpr_archive_info = None |
- page_set_archive_info.AssertValidCloudStorageBucket(cloud_storage_bucket) |
+ archive_info.AssertValidCloudStorageBucket(cloud_storage_bucket) |
self._cloud_storage_bucket = cloud_storage_bucket |
self._base_dir = os.path.dirname(inspect.getfile(self.__class__)) |
@@ -54,9 +54,8 @@ class UserStorySet(object): |
def wpr_archive_info(self): |
"""Lazily constructs wpr_archive_info if it's not set and returns it.""" |
if self.archive_data_file and not self._wpr_archive_info: |
- self._wpr_archive_info = ( |
- page_set_archive_info.PageSetArchiveInfo.FromFile( |
- os.path.join(self.base_dir, self.archive_data_file), self.bucket)) |
+ self._wpr_archive_info = archive_info.WprArchiveInfo.FromFile( |
+ os.path.join(self.base_dir, self.archive_data_file), self.bucket) |
return self._wpr_archive_info |
def AddUserStory(self, user_story): |