[system-health] Unify SH stories interface

tools/perf/benchmarks/battor.py

Index: tools/perf/benchmarks/battor.py
diff --git a/tools/perf/benchmarks/battor.py b/tools/perf/benchmarks/battor.py
index 73d8ae301781d60a0ccb79e9b2cc6cb093a1c9b2..b73963d425cbc062aa03d74c75f59ddf1987b332 100644
--- a/tools/perf/benchmarks/battor.py
+++ b/tools/perf/benchmarks/battor.py
@@ -55,7 +55,8 @@ class BattOrSystemHealthLoadingDesktop(_BattOrBenchmark):
   """Desktop Chrome Memory System Health Benchmark."""
 
   def CreateStorySet(self, options):
-    return page_sets.DesktopSystemHealthStorySet(take_memory_measurement=False)
+    return page_sets.SystemHealthStorySet(platform='desktop',
+                                          story_name_regex=r'load:.*')
 
   @classmethod
   def ShouldDisable(cls, possible_browser):
@@ -71,7 +72,8 @@ class BattOrSystemHealthLoadingMobile(_BattOrBenchmark):
   """Mobile Chrome Memory System Health Benchmark."""
 
   def CreateStorySet(self, options):
-    return page_sets.MobileSystemHealthStorySet(take_memory_measurement=False)
+    return page_sets.SystemHealthStorySet(platform='mobile',
+                                          story_name_regex=r'load:.*')
 
   @classmethod
   def ShouldDisable(cls, possible_browser):