upgrade perf code

1 files changed, 3 insertions, 2 deletions

diff --git a/scripts/internal/bench_oneshot_2.py b/scripts/internal/bench_oneshot_2.py
index b5758149..f61aca99 100644
--- a/scripts/internal/bench_oneshot_2.py
+++ b/scripts/internal/bench_oneshot_2.py
@@ -11,7 +11,7 @@ supposed to be more precise.
 
 import sys
 
-import perf.text_runner
+from perf import Runner
 
 import psutil
 from bench_oneshot import names
@@ -37,7 +37,7 @@ def prepare_cmd(runner, cmd):
 
 
 def main():
-    runner = perf.text_runner.TextRunner(name='psutil')
+    runner = Runner()
     runner.argparser.add_argument('benchmark', choices=('normal', 'oneshot'))
     runner.prepare_subprocess_args = prepare_cmd
 
@@ -53,4 +53,5 @@ def main():
     else:
         runner.bench_func(call_oneshot, funs)
 
+
 main()