test keylookup

biothings · Oct 3, 2024 · 97c75b3 · 97c75b3
1 parent 8d8e568
commit 97c75b3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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):