Commit 606f633 1 parent a020f76 commit 606f633 Copy full SHA for 606f633
File tree 1 file changed +7
-1
lines changed
1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change 16
16
from rich .table import Table
17
17
from rich .text import Text
18
18
19
- from kaprese .core .benchmark import Benchmark
19
+ from kaprese .core .benchmark import Benchmark , all_benchmarks
20
20
from kaprese .core .engine import Engine
21
21
from kaprese .core .runner import Runner
22
22
from kaprese .utils .console import PanelConsole , console
@@ -321,6 +321,12 @@ def main(
321
321
continue
322
322
engines .append (engine )
323
323
benchmarks : list [Benchmark ] = []
324
+
325
+ # Load benchmarks
326
+ # If no benchmark is specified, run all benchmarks
327
+ if len (args .benchmark ) == 0 :
328
+ benchmarks = all_benchmarks ()
329
+
324
330
for bench_name in args .benchmark :
325
331
bench = Benchmark .load (bench_name )
326
332
if bench is None :
You can’t perform that action at this time.
0 commit comments