Merge pull request #4 from biothings/self-edges

Filter self-descendants
biothings · Dec 5, 2023 · e4bb786 · e4bb786
2 parents aed33db + 6285f39
commit e4bb786
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -40,10 +40,16 @@ exports.getDescendants = (curies, recursive = true) => {
         }
         level = next_level;
       }
-      children[curie] = _.uniq(children[curie]).slice(0, ENTITY_CAP);
+      children[curie] = _.uniq(children[curie])
+        .filter(child => child !== curie)
+        .slice(0, ENTITY_CAP);
     }
     return children;
   } else {
-    return _.pick(data, curies);
+    return Object.fromEntries(
+      Object.entries(_.pick(data, curies)).map(([curie, descendants]) => {
+        return [curie, descendants.filter(descendant => descendant !== curie)];
+      }),
+    );
   }
 };