3698e34abfd6215605a21f28bb61dba872c5cc63,nlp_architect/data/cdc_resources/relations/referent_dict_relation_extraction.py,ReferentDictRelationExtraction,__init__,#ReferentDictRelationExtraction#,31

Before Change

            self.ref_dict = load_json_file(ref_dict)
        elif method == OnlineOROfflineMethod.ONLINE:
            self.ref_dict = self.load_reference_dict(ref_dict)
        logger.info("ReferentDict module lead successfully")
        super(ReferentDictRelationExtraction, self).__init__()

    def extract_all_relations(self, mention_x: MentionDataLight,
                              mention_y: MentionDataLight) -> Set[RelationType]:

After Change

            ref_dict (required): str Location of referent dictionary file to work with
        
        logger.info("Loading ReferentDict module")
        if ref_dict is not None and os.path.isfile(ref_dict):
            if method == OnlineOROfflineMethod.OFFLINE:
                self.ref_dict = load_json_file(ref_dict)
            elif method == OnlineOROfflineMethod.ONLINE:
                self.ref_dict = self.load_reference_dict(ref_dict)
            logger.info("ReferentDict module lead successfully")
        else:
            raise FileNotFoundError("Referent Dict file not found or not in path..")

        super(ReferentDictRelationExtraction, self).__init__()

    def extract_all_relations(self, mention_x: MentionDataLight,
                              mention_y: MentionDataLight) -> Set[RelationType]:

Instances