Fix validation error on Triple vs ChemicalToDiseaseRelationship

monarch-initiative · Oct 3, 2023 · f9ef356 · f9ef356
1 parent 7d84e89
commit f9ef356
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ontogpt/evaluation/ctd/eval_ctd.py b/src/ontogpt/evaluation/ctd/eval_ctd.py
@@ -159,7 +159,7 @@ def load_cases(self, path: Path) -> Iterable[ChemicalToDiseaseDocument]:
                 # text = f"Title: {title} Abstract: {abstract}"
                 for r in document.relations:
                     i = r.infons
-                    t = Triple.model_validate(
+                    t = ChemicalToDiseaseRelationship.model_validate(
                         {
                             "subject": f"{self.subject_prefix}:{i['Chemical']}",
                             "predicate": RMAP[i["relation"]],
@@ -169,10 +169,10 @@ def load_cases(self, path: Path) -> Iterable[ChemicalToDiseaseDocument]:
                     triples_by_text[(title, abstract)].append(t)
         i = 0
         for (title, abstract), triples in triples_by_text.items():
-            i = str(i + 1)
+            i = i + 1
             pub = Publication.model_validate(
                 {
-                    "id": i,
+                    "id": str(i),
                     "title": title,
                     "abstract": abstract,
                     "combined_text": "N/A",