Skip to content
Snippets Groups Projects
Commit dd7a89ba authored by Florian Fischer's avatar Florian Fischer
Browse files

load benchmarks using importlib instead of eval

parent c66f450a
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment