CORE-55: stop building per-call gamla compositions in neighbors/run_on_kg_and_node

knowledge_graph/querying_raw.py

@@ -18,12 +18,13 @@ def neighbors_reversed(
     graph: triplets_index.TripletsWithIndex,
     node: triplet.Element,
 ) -> FrozenSet[triplet.Element]:
-    return gamla.pipe(
-        triplets_index.retrieve(relation, triplets_index.object_relation_index)(
-            graph, node
-        ),
-        gamla.map(triplet.subject),
-        frozenset,
+    return frozenset(
+        map(
+            triplet.subject,
+            triplets_index.retrieve(relation, triplets_index.object_relation_index)(
+                graph, node
+            ),
+        )
     )
 
 
@@ -33,12 +34,13 @@ def neighbors(
     graph: triplets_index.TripletsWithIndex,
     node: triplet.Element,
 ) -> FrozenSet[triplet.Element]:
-    return gamla.pipe(
-        triplets_index.retrieve(relation, triplets_index.subject_relation_index)(
-            graph, node
-        ),
-        gamla.map(triplet.object),
-        frozenset,
+    return frozenset(
+        map(
+            triplet.object,
+            triplets_index.retrieve(relation, triplets_index.subject_relation_index)(
+                graph, node
+            ),
+        )
     )
 
 

knowledge_graph/storage.py

@@ -71,12 +71,11 @@ def inner(node: Node):
 
 
 def run_on_kg_and_node(f):
+    f_any_output = run_on_kg_and_node_any_output(f)
+
     def inner(node: Node) -> Nodes:
-        return gamla.pipe(
-            node,
-            run_on_kg_and_node_any_output(f),
-            gamla.map(lambda result_node: Node(node.graph_id, result_node)),
-            frozenset,
+        return frozenset(
+            Node(node.graph_id, result_node) for result_node in f_any_output(node)
         )
 
     return inner

pyproject.toml

@@ -4,14 +4,15 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "knowledge-graph"
-version = "0.0.34"
+version = "0.0.35"
 description = "A library to store data in a knowledge graph"
 readme = "README.md"
 requires-python = ">=3.14"
 authors = [{ name = "Hyro AI", email = "contact@hyro.ai" }]
 keywords = ["tag1", "tag2"]
 dependencies = [
     "dataclasses_json==0.5.7",
+    "gamla==164",
     "immutables",
     "phonenumbers",
     "pytest",