def perform_inference()

in data_preparation/model_utils.py [0:0]

• 7 lines of code
• 1 McCabe index (conditional complexity)

def perform_inference(concrete_fn: Callable) -> Callable:
    def fn(image: np.ndarray) -> Image.Image:
        preprocessed_image = image_utils.preprocess_image(image)
        result = concrete_fn(preprocessed_image)["final_output:0"]
        output_image = image_utils.postprocess_image(result)
        return output_image

    return fn