diff --git a/src/hub/dataload/sources/chembl/parser.py b/src/hub/dataload/sources/chembl/parser.py index 91dff82..15d9af3 100644 --- a/src/hub/dataload/sources/chembl/parser.py +++ b/src/hub/dataload/sources/chembl/parser.py @@ -20,6 +20,7 @@ def parse_data(data): uniprot_accessions.append(accession) if uniprot_accessions: output = { + "_id": item["target_chembl_id"], "chembl_target": item["target_chembl_id"], "xrefs": { "accession": uniprot_accessions, diff --git a/src/hub/dataload/sources/chembl/upload.py b/src/hub/dataload/sources/chembl/upload.py index 2281b6d..acad8ee 100644 --- a/src/hub/dataload/sources/chembl/upload.py +++ b/src/hub/dataload/sources/chembl/upload.py @@ -26,9 +26,9 @@ def load_data(self, data_folder): target_filepaths = glob.iglob( os.path.join(data_folder, self.TARGET_FILENAME_PATTERN) ) - for doc in load_data(target_filepaths): - yield doc - # return self.keylookup(load_data(target_filepaths)) + # for doc in load_data(target_filepaths): + # yield doc + return self.keylookup(load_data(target_filepaths)) @classmethod def get_mapping(klass):