Skip to content

Commit a8ae171

Browse files
committed
修复两处Bug:
1. 修复eval.py未指定导出文件路径时的报错。 2. 修复eval.py同时未指定include_*和exclude_*的时候的报错,这种情况默认使用json文件读取后字典的键的列表。
1 parent fa51d1e commit a8ae171

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

eval.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -119,10 +119,14 @@ def get_args():
119119
)
120120
args = parser.parse_args()
121121

122-
os.makedirs(os.path.dirname(args.metric_npy), exist_ok=True)
123-
os.makedirs(os.path.dirname(args.curves_npy), exist_ok=True)
124-
os.makedirs(os.path.dirname(args.record_txt), exist_ok=True)
125-
os.makedirs(os.path.dirname(args.record_xlsx), exist_ok=True)
122+
if args.metric_npy:
123+
os.makedirs(os.path.dirname(args.metric_npy), exist_ok=True)
124+
if args.curves_npy:
125+
os.makedirs(os.path.dirname(args.curves_npy), exist_ok=True)
126+
if args.record_txt:
127+
os.makedirs(os.path.dirname(args.record_txt), exist_ok=True)
128+
if args.record_xlsx:
129+
os.makedirs(os.path.dirname(args.record_xlsx), exist_ok=True)
126130
if args.to_overwrite and not args.record_txt:
127131
warnings.warn("--to-overwrite only works with a valid --record-txt")
128132
return args

utils/generate_info.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,9 @@ def get_valid_elements(source: list, include_elements: list, exclude_elements: l
6868
if element not in exclude_elements:
6969
targeted_elements.append(element)
7070

71+
elif not include_elements and not exclude_elements:
72+
targeted_elements = source
73+
7174
else:
7275
raise ValueError(
7376
f"include_elements: {include_elements}\nexclude_elements: {exclude_elements}"

0 commit comments

Comments
 (0)