From 843006c27f9b760d541596e8829a6ebb9a66df6d Mon Sep 17 00:00:00 2001
From: uhbrar <abrar.mesbah@gmail.com>
Date: Mon, 15 Apr 2024 12:53:20 -0400
Subject: [PATCH] update omnicorp set handling

---
 ranker/modules/omnicorp_overlay.py | 5 +----
 requirements.txt                   | 2 +-
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/ranker/modules/omnicorp_overlay.py b/ranker/modules/omnicorp_overlay.py
index c0e879b..8e3a2ef 100644
--- 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
index 0886795..b3c269b 100644
--- 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