diff --git a/test/api_unittest.py b/test/api_unittest.py index d16ea934e12bb7c222f2716dee030bf3a1063899..73b57b7d102c15eecae807ec57a28eec2e0d2912 100644 --- a/test/api_unittest.py +++ b/test/api_unittest.py @@ -10,7 +10,8 @@ from otbtf.examples.tensorflow_v2x.fcnn import train_from_tfrecords from otbtf.examples.tensorflow_v2x.fcnn.fcnn_model import INPUT_NAME, \ OUTPUT_SOFTMAX_NAME, OUTPUT_ARGMAX_NAME, TARGET_NAME from otbtf.model import cropped_tensor_name -from test_utils import resolve_paths, files_exist, run_command_and_compare +from test_utils import resolve_paths, files_exist, run_command_and_compare, \ + run_command_and_test_exist INFERENCE_MAE_TOL = 10.0 # Dummy value: we don't really care of the mae value but rather the image size etc