diff --git a/src/ontogpt/evaluation/ctd/eval_ctd.py b/src/ontogpt/evaluation/ctd/eval_ctd.py index feb597f4c..340ea427a 100644 --- 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):