Merge pull request #42 from GeneDx/genSimFix

fix #41
GeneDx · Dec 29, 2020 · 5280101 · 5280101
2 parents 21903a7 + bc69171
commit 5280101
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/txt2hpo/nlp.py b/txt2hpo/nlp.py
@@ -72,7 +72,7 @@ def similarity_term_to_context(term, context, model):
     :return: float
     """
     def remove_out_of_vocab(tokens):
-        return [x for x in tokens if x in model.vocab]
+        return [x for x in tokens if x in model.key_to_index]
 
     hpo_term = hpo_network.nodes[term]
     hpo_term_definition = hpo_term['name']