load benchmarks using importlib instead of eval

1 files changed, 5 insertions, 1 deletions

diff --git a/bench.py b/bench.py
index eda6988..6360db5 100755
--- a/bench.py
+++ b/bench.py
@@ -203,7 +203,11 @@ def main():
             os.makedirs(bench_res_dir, exist_ok=True)
 
         try:
-            bench = eval(f"importlib.import_module('src.benchmarks.{bench}').{bench}")
+            bench_module = importlib.import_module(f"src.benchmarks.{bench}")
+            if not hasattr(bench_module, bench):
+                continue
+
+            bench = getattr(bench_module, bench)
 
             if args.load:
                 bench.load(path=args.load)