Allow setting the amount of warmup values in Runner

psf · Jul 16, 2024 · 168bc95 · 168bc95
1 parent 7b9a23a
commit 168bc95
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyperf/_runner.py b/pyperf/_runner.py
@@ -75,7 +75,7 @@ def __init__(self, values=None, processes=None,
                  loops=0, min_time=0.1, metadata=None,
                  show_name=True,
                  program_args=None, add_cmdline_args=None,
-                 _argparser=None):
+                 _argparser=None, warmups=1):
 
         # Reset the stats collection if running a --enable-pystats build
         try:
@@ -165,7 +165,7 @@ def __init__(self, values=None, processes=None,
                             help='number of values per process (default: %s)'
                                  % values)
         parser.add_argument('-w', '--warmups',
-                            type=positive_or_nul,
+                            type=positive_or_nul, default=warmups,
                             help='number of skipped values per run used '
                                  'to warmup the benchmark')
         parser.add_argument('-l', '--loops',