def convert_ds_element_to_tuple(element):
    input_keys = ['image', 'horz_split_points_probs', 'vert_split_points_probs', 'horz_split_points_binary', 'vert_split_points_binary']
    return ({key: element[key] for key in input_keys}, {'markup_table': element['markup_table']})
def convert_ds_element_to_tuple(element):
    input_keys = ['image', 'horz_split_points_probs', 'vert_split_points_probs', 'horz_split_points_binary', 'vert_split_points_binary']
    return ({key: element[key] for key in input_keys}, {'markup_table': element['markup_table']})