def identity_transform(memory, image_key):
    """Simple function to return the image at "image_key" in the memory.

    Args:
        memory (dictionnary): memory dict.
        image_key (string): image key.

    Returns:
        np.array: the image.
    """
    return memory[image_key]