diff --git a/opencompass/__init__.py b/opencompass/__init__.py index 0404d810..e1424ed0 100644 --- a/opencompass/__init__.py +++ b/opencompass/__init__.py @@ -1 +1 @@ -__version__ = '0.3.0' +__version__ = '0.3.1' diff --git a/opencompass/utils/run.py b/opencompass/utils/run.py index 8e7cf093..dfcca692 100644 --- a/opencompass/utils/run.py +++ b/opencompass/utils/run.py @@ -115,7 +115,7 @@ def get_config_from_arg(args) -> Config: datasets = [] if args.datasets: script_dir = os.path.dirname(os.path.abspath(__file__)) - parent_dir = os.path.dirname(os.path.dirname(script_dir)) + parent_dir = os.path.dirname(script_dir) default_configs_dir = os.path.join(parent_dir, 'configs') datasets_dir = [ os.path.join(args.config_dir, 'datasets'), @@ -154,7 +154,7 @@ def get_config_from_arg(args) -> Config: raise ValueError('You must specify a config file path, or specify --models and --datasets, or specify HuggingFace model parameters and --datasets.') models = [] script_dir = os.path.dirname(os.path.abspath(__file__)) - parent_dir = os.path.dirname(os.path.dirname(script_dir)) + parent_dir = os.path.dirname(script_dir) default_configs_dir = os.path.join(parent_dir, 'configs') models_dir = [ os.path.join(args.config_dir, 'models'), @@ -197,7 +197,7 @@ def get_config_from_arg(args) -> Config: # parse summarizer args summarizer_arg = args.summarizer if args.summarizer is not None else 'example' script_dir = os.path.dirname(os.path.abspath(__file__)) - parent_dir = os.path.dirname(os.path.dirname(script_dir)) + parent_dir = os.path.dirname(script_dir) default_configs_dir = os.path.join(parent_dir, 'configs') summarizers_dir = [ os.path.join(args.config_dir, 'summarizers'),