avoid repeated work

graphcore-research · Oct 16, 2024 · ea845ab · ea845ab
1 parent c49e886
commit ea845ab
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/kg_topology_toolbox/topology_toolbox.py b/src/kg_topology_toolbox/topology_toolbox.py
@@ -547,10 +547,12 @@ def edge_pattern_summary(
 
         # composition & metapaths
         if return_metapath_list:
-            counts = self.edge_metapath_count(
-                filter_relations,
-                composition_chunk_size,
-                composition_workers,
+            counts = composition_count(
+                df_triangles,
+                chunk_size=composition_chunk_size,
+                workers=composition_workers,
+                metapaths=True,
+                directed=True,
             )
             counts["metapath"] = (
                 counts["r1"].astype(str) + "-" + counts["r2"].astype(str)