File tree Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Original file line number Diff line number Diff 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 ))
Original file line number Diff line number Diff line change 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" ,
You can’t perform that action at this time.
0 commit comments