def input_fn()

in src/similarity/batch.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

def input_fn(request_body, accept=INPUT_CONTENT_TYPE):
    logger.info('Deserializing the generated input.')
    if accept == INPUT_CONTENT_TYPE:

      #  logger.info(request_body)
        logger.info(len(request_body))
        logger.info(pickle.format_version)
        logger.info(np.version.version)

        request_body = gzip.decompress(request_body)
        (names, tensors) = pickle.load(BytesIO(request_body), fix_imports=True)
        global image_names
        image_names = names
        
        logger.info(tensors.shape)
        logger.info(image_names)
        return torch.from_numpy(tensors)
    raise Exception('Requested unsupported ContentType: ' + accept)