fix: add cache for local KG

IBCNServices · Apr 9, 2021 · 7b1a41e · 7b1a41e
1 parent 309d1aa
commit 7b1a41e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pyrdf2vec/graphs/kg.py b/pyrdf2vec/graphs/kg.py
@@ -406,6 +406,9 @@ def _fill_hops(self, entities: Entities) -> None:
             hops = self._res2hops(Vertex(entity), res)
             self._entity_hops.update({entity: hops})
 
+    @cachedmethod(
+        operator.attrgetter("cache"), key=partial(hashkey, "_get_hops")
+    )
     def _get_hops(self, vertex: Vertex, is_reverse: bool = False) -> List[Hop]:
         """Returns the hops of a vertex for a local Knowledge Graph.