mend

mims-harvard · Jan 20, 2025 · b237dd6 · b237dd6
1 parent 6ef92ec
commit b237dd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tdc/model_server/tokenizers/geneformer.py b/tdc/model_server/tokenizers/geneformer.py
@@ -73,7 +73,7 @@ def tokenize_cell_vectors(self,
         ])
         coding_miRNA_ids = adata.var[ensembl_id][coding_miRNA_loc]
         coding_miRNA_tokens = np.array(
-            [self.gene_token_dict[i] for i in coding_miRNA_ids])
+            [self.gene_token_dict.get(i, 0) for i in coding_miRNA_ids])
 
         try:
             _ = adata.obs["filter_pass"]