update omnicorp set handling

ranking-agent · Apr 15, 2024 · 843006c · 843006c
1 parent ad04130
commit 843006c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/ranker/modules/omnicorp_overlay.py b/ranker/modules/omnicorp_overlay.py
@@ -207,10 +207,7 @@ async def query(request: PDResponse):
             [
                 n
                 for n in qgraph["nodes"]
-                if (
-                    ("is_set" in qgraph["nodes"][n])
-                    and (qgraph["nodes"][n]["is_set"])
-                )
+                if (qgraph["nodes"][n].get("set_interpretation", None) or "BATCH") != "BATCH"
             ]
         )
 

diff --git a/requirements.txt b/requirements.txt
@@ -11,5 +11,5 @@ uvicorn==0.17.6
 uvloop==0.19.0
 numpy==1.26.4
 httptools==0.6.0
-reasoner-pydantic==5.0.1
+reasoner-pydantic==5.0.2
 yappi==1.6.0