@@ -877,7 +877,7 @@ def otx_explain_testing(template, root, otx_dir, args, trained=False):
877
877
assert os .path .exists (output_dir )
878
878
if trained :
879
879
assert len (os .listdir (output_dir )) > 0
880
- assert all ([os .path .splitext (fname )[1 ] == ".tiff" for fname in os .listdir (output_dir )])
880
+ assert all ([os .path .splitext (fname )[1 ] in [ ".tiff" , ".log" ] for fname in os .listdir (output_dir )])
881
881
882
882
883
883
def otx_explain_testing_all_classes (template , root , otx_dir , args ):
@@ -964,7 +964,7 @@ def otx_explain_testing_process_saliency_maps(template, root, otx_dir, args, tra
964
964
assert os .path .exists (output_dir )
965
965
if trained :
966
966
assert len (os .listdir (output_dir )) > 0
967
- assert all ([os .path .splitext (fname )[1 ] == ".png" for fname in os .listdir (output_dir )])
967
+ assert all ([os .path .splitext (fname )[1 ] in [ ".png" , ".log" ] for fname in os .listdir (output_dir )])
968
968
969
969
970
970
def otx_explain_openvino_testing (template , root , otx_dir , args , trained = False ):
@@ -1005,7 +1005,7 @@ def otx_explain_openvino_testing(template, root, otx_dir, args, trained=False):
1005
1005
assert os .path .exists (output_dir )
1006
1006
if trained :
1007
1007
assert len (os .listdir (output_dir )) > 0
1008
- assert all ([os .path .splitext (fname )[1 ] == ".tiff" for fname in os .listdir (output_dir )])
1008
+ assert all ([os .path .splitext (fname )[1 ] in [ ".tiff" , ".log" ] for fname in os .listdir (output_dir )])
1009
1009
1010
1010
1011
1011
def otx_explain_all_classes_openvino_testing (template , root , otx_dir , args ):
@@ -1094,7 +1094,7 @@ def otx_explain_process_saliency_maps_openvino_testing(template, root, otx_dir,
1094
1094
assert os .path .exists (output_dir )
1095
1095
if trained :
1096
1096
assert len (os .listdir (output_dir )) > 0
1097
- assert all ([os .path .splitext (fname )[1 ] == ".png" for fname in os .listdir (output_dir )])
1097
+ assert all ([os .path .splitext (fname )[1 ] in [ ".png" , ".log" ] for fname in os .listdir (output_dir )])
1098
1098
1099
1099
1100
1100
def otx_find_testing ():
0 commit comments