Skip to content

Commit f74d017

Browse files
authored
Merge branch 'master' into master
2 parents 9361db4 + c570ff4 commit f74d017

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

tools/submission/preprocess_submission.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,6 @@ def delete_empty_dirs(src):
8484
"""
8585
if not os.path.isdir(src):
8686
return False
87-
8887
if all([delete_empty_dirs(os.path.join(src, file))
8988
for file in os.listdir(src)]):
9089
log.info("Removing empty dir: (%s)", src)
@@ -532,9 +531,11 @@ def main():
532531
if not args.nodelete_empty_dirs:
533532
delete_empty_dirs(os.path.join(src_dir))
534533

534+
run_dir = os.getcwd()
535535
os.chdir(src_dir)
536536

537537
infer_scenario_results(args, config)
538+
os.chdir(run_dir)
538539

539540
if not args.nodelete_empty_dirs:
540541
delete_empty_dirs(os.path.join(src_dir))

tools/submission/submission_checker.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,7 @@
188188
"sample_index_rng_seed": 198141574272810017,
189189
"schedule_rng_seed": 7575108116881280410,
190190
},
191+
# not required for v5.0+
191192
"test05_seeds": {
192193
# TODO: Update random seeds
193194
"qsl_rng_seed": 2376919268182438552,
@@ -2880,7 +2881,7 @@ def check_compliance_dir(
28802881
compliance_perf_pass = True
28812882
compliance_perf_dir_pass = True
28822883
compliance_acc_pass = True
2883-
test_list = ["TEST01", "TEST04", "TEST05"]
2884+
test_list = ["TEST01", "TEST04"]
28842885

28852886
if model in [
28862887
"bert-99",
@@ -2899,15 +2900,15 @@ def check_compliance_dir(
28992900
]:
29002901
test_list.remove("TEST04")
29012902

2902-
if model in [
2903+
if config.version in ["v4.0", "v4.1"] and model not in [
29032904
"gptj-99",
29042905
"gptj-99.9",
29052906
"llama2-70b-99",
29062907
"llama2-70b-99.9",
29072908
"stable-diffusion-xl",
29082909
"mixtral-8x7b",
2909-
] or config.version not in ["v4.0", "v4.1"]:
2910-
test_list.remove("TEST05")
2910+
]:
2911+
test_list.append("TEST05")
29112912

29122913
if model in [
29132914
"gptj-99",

0 commit comments

Comments
 (0)