hardcode treats predicate

RobokopU24 · May 17, 2024 · b17fe6a · b17fe6a
1 parent 36bc954
commit b17fe6a
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/src/stores/useBiolinkModel.js b/src/stores/useBiolinkModel.js
@@ -28,15 +28,15 @@ export default function useBiolinkModel() {
    * @returns {object[]} list of predicate objects
    */
   function getEdgePredicates() {
-    const newPredicates =
-      Object.entries(biolinkModel.slots)
-        .filter(checkIfDescendantOfRelatedTo)
-        .map(([identifier, predicate]) => ({
-          predicate: strings.edgeFromBiolink(identifier),
-          domain: strings.nodeFromBiolink(predicate.domain),
-          range: strings.nodeFromBiolink(predicate.range),
-        }));
-    return newPredicates;
+    const newPredicates = Object.entries(biolinkModel.slots).filter(checkIfDescendantOfRelatedTo);
+    // hard code in treats, it's techincally not a descendant of `related to`
+    // TODO: we'll want the more correct parsing using mixins at some point
+    newPredicates.push(['treats', biolinkModel.slots.treats]);
+    return newPredicates.map(([identifier, predicate]) => ({
+      predicate: strings.edgeFromBiolink(identifier),
+      domain: strings.nodeFromBiolink(predicate.domain),
+      range: strings.nodeFromBiolink(predicate.range),
+    }));
   }
 
   /**