fix: update bucket map to use environment variables for default values

chandra-siri · chandra-siri · commit 1298818ee1db · 2026-01-15T18:06:47.000Z
diff --git a/tests/perf/microbenchmarks/reads/config.py b/tests/perf/microbenchmarks/reads/config.py
@@ -44,8 +44,8 @@ def _get_params() -> Dict[str, List[ReadParameters]]:
     rounds = common_params["rounds"]
 
     bucket_map = {
-        "zonal": config["defaults"]["DEFAULT_RAPID_ZONAL_BUCKET"],
-        "regional": config["defaults"]["DEFAULT_STANDARD_BUCKET"],
+        "zonal": os.environ.get("DEFAULT_RAPID_ZONAL_BUCKET", config['defaults']['DEFAULT_RAPID_ZONAL_BUCKET']),
+        "regional": os.environ.get("DEFAULT_STANDARD_BUCKET", config['defaults']['DEFAULT_STANDARD_BUCKET'])
     }
 
     for workload in config["workload"]:

Original file line number	Diff line number	Diff line change
`@@ -44,8 +44,8 @@ def _get_params() -> Dict[str, List[ReadParameters]]:`
`44`	`44`	`rounds = common_params["rounds"]`
`45`	`45`
`46`	`46`	`bucket_map = {`
`47`		`- "zonal": config["defaults"]["DEFAULT_RAPID_ZONAL_BUCKET"],`
`48`		`- "regional": config["defaults"]["DEFAULT_STANDARD_BUCKET"],`
	`47`	`+ "zonal": os.environ.get("DEFAULT_RAPID_ZONAL_BUCKET", config['defaults']['DEFAULT_RAPID_ZONAL_BUCKET']),`
	`48`	`+ "regional": os.environ.get("DEFAULT_STANDARD_BUCKET", config['defaults']['DEFAULT_STANDARD_BUCKET'])`
`49`	`49`	`}`
`50`	`50`
`51`	`51`	`for workload in config["workload"]:`