diff options
Diffstat (limited to 'scripts')
| -rwxr-xr-x | scripts/print_facts.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/scripts/print_facts.py b/scripts/print_facts.py index ec9404c..6099b6f 100755 --- a/scripts/print_facts.py +++ b/scripts/print_facts.py @@ -20,17 +20,15 @@ import argparse import importlib -import inspect import os import sys -CURRENTDIR = os.path.dirname( - os.path.abspath(inspect.getfile(inspect.currentframe()))) +CURRENTDIR = os.path.dirname(os.path.abspath(__file__)) PARENTDIR = os.path.dirname(CURRENTDIR) sys.path.insert(0, PARENTDIR) +from allocbench.benchmark import AVAIL_BENCHMARKS import allocbench.facter as facter -from allocbench.globalvars import BENCHMARKS from allocbench.plots import print_facts, print_common_facts from allocbench.util import print_error @@ -49,7 +47,7 @@ def main(): cwd = os.getcwd() os.chdir(args.results) - for benchmark in BENCHMARKS: + for benchmark in AVAIL_BENCHMARKS: bench_module = importlib.import_module( f"allocbench.benchmarks.{benchmark}") |