fix: filter self-descendants

biothings · Nov 28, 2023 · 6285f39 · 6285f39
1 parent aed33db
commit 6285f39
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)];
+      }),
+    );
   }
 };