fix job crash (#57)

* fix job crash * try with latest py2neo version
stewart-lab · May 24, 2024 · 50ddb11 · 50ddb11
1 parent 22d5c8a
commit 50ddb11
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -14,5 +14,5 @@ flask-bcrypt==0.7.1
 pymongo==4.0.1
 Werkzeug==2.0.2
 neo4j==4.4.5
-py2neo==2021.2.3
+py2neo
 pure-cdb
diff --git a/src/workers/work.py b/src/workers/work.py
@@ -106,7 +106,9 @@ def km_work_all_vs_all(json: dict):
 
         while b_term_set:
             b_term = li.the_index.get_highest_priority_term(b_term_set, b_term_token_dict)
-            b_term_set.remove(b_term)
+
+            if b_term in b_term_set:
+                b_term_set.remove(b_term)
 
             res = km.kinderminer_search(a_term, b_term, li.the_index, censor_year, return_pmids, top_n_articles, scoring)
 
@@ -153,7 +155,9 @@ def km_work_all_vs_all(json: dict):
 
         while c_term_set:
             c_term = li.the_index.get_highest_priority_term(c_term_set, c_term_token_dict)
-            c_term_set.remove(c_term)
+
+            if c_term in c_term_set:
+                c_term_set.remove(c_term)
 
             for ab in ab_results:
                 abc_result = {