Merge pull request #396 from UnsignedByte/brench-globarr

Allow for array of glob patterns in brench
sampsyo · Feb 11, 2025 · 90014fe · 90014fe
2 parents d0967af + e3d7ba5
commit 90014fe
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/brench/brench.py b/brench/brench.py
@@ -87,8 +87,11 @@ def brench(config_path, files, jobs):
         config = tomlkit.loads(f.read())
 
     # Use configured file list, if none is specified via the CLI.
-    if not files and 'benchmarks' in config:
-        files = glob.glob(config['benchmarks'], recursive=True)
+    if not files and "benchmarks" in config:
+        files = config["benchmarks"]
+        if isinstance(files, str):
+            files = [files]
+        files = sum([glob.glob(f, recursive=True) for f in files], [])
 
     timeout = config.get('timeout', 5)