def generate_data_sample(
            doc: List, ent1: int, ent2: int, label: List, ent_id2ent: Dict, ent_id2ent_tag: Dict
    ) -> Tuple[List[Union[List, List]], List]:
        """ Creates an entry of processed docred corpus """
        return (
                    [
                        doc,
                        [ent_id2ent[ent1], ent_id2ent[ent2]],
                        [ent_id2ent_tag[ent1], ent_id2ent_tag[ent2]]
                    ],
                    label
                )
 Creates an entry of processed docred corpus 
deeppavlov/dataset_readers/docred_reader.py