Try with config file

benchmarks/benchmarks/brainglobe_atlasapi.py

@@ -1,3 +1,7 @@
+import json
+import os
+from pathlib import Path
+
 from brainglobe_workflows.brainglobe_atlasapi.create_mouse_atlas import (
     create_mouse_atlas,
 )
@@ -6,14 +10,27 @@
 class TimeBenchmark:
     # Timing attributes
     timeout = 3600  # default: 60 s
-    version = (
-        None  # benchmark version. Default:None (i.e. hash of source code)
-    )
-    warmup_time = 0.1  # seconds
-    rounds = 2
-    repeat = 0
-    sample_time = 0.01  # default: 10 ms = 0.01 s;
-    min_run_count = 2  # default:2
+
+    def setup(self):
+        # read input config: environment variable if it exists, else default
+        input_config_path = os.getenv(
+            "ATLAS_CONFIG_PATH",
+            default=str(
+                Path(__file__).parents[2]
+                / "brainglobe_workflows"
+                / "configs"
+                / "brainglobe_atlasapi_small.json"
+            ),
+        )
+
+        assert Path(input_config_path).exists()
+
+        # read as dict
+        with open(input_config_path) as cfg:
+            config_dict = json.load(cfg)
+
+        # pass dict to class
+        self.config = config_dict
 
     def time_create_mouse_atlas(self):
-        create_mouse_atlas()
+        create_mouse_atlas(**self.config)

brainglobe_workflows/brainglobe_atlasapi/create_mouse_atlas.py

@@ -1,10 +1,10 @@
 from brainglobe_atlasapi import BrainGlobeAtlas
 
 
-def create_mouse_atlas():
-    atlas = BrainGlobeAtlas("allen_mouse_100um")
+def create_mouse_atlas(atlas_name):
+    atlas = BrainGlobeAtlas(atlas_name)
     print(atlas.atlas_name)
 
 
 if __name__ == "__main__":
-    create_mouse_atlas()
+    create_mouse_atlas("allen_mouse_100um")

brainglobe_workflows/configs/brainglobe_atlasapi_large.json

@@ -0,0 +1,3 @@
+{
+    "atlas_name": "allen_mouse_10um"
+}

brainglobe_workflows/configs/brainglobe_atlasapi_small.json

@@ -0,0 +1,3 @@
+{
+    "atlas_name": "allen_mouse_100um"
+}