Repair negation

monarch-initiative · Oct 3, 2023 · 7b38c39 · 7b38c39
1 parent 37e2d6c
commit 7b38c39
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/ontogpt/evaluation/ctd/eval_ctd.py b/src/ontogpt/evaluation/ctd/eval_ctd.py
@@ -37,7 +37,6 @@
     ChemicalToDiseaseRelationship,
     Publication,
     TextWithTriples,
-    Triple,
 )
 
 THIS_DIR = Path(__file__).parent
@@ -50,8 +49,11 @@
 logger = logging.getLogger(__name__)
 
 
-def negated(Triple) -> bool:
-    return Triple.qualifier and Triple.qualifier.lower() == "not"
+def negated(ChemicalToDiseaseRelationship) -> bool:
+    return (
+        ChemicalToDiseaseRelationship.qualifier
+        and ChemicalToDiseaseRelationship.qualifier.lower() == "not"
+    )
 
 
 class PredictionRE(BaseModel):