📝 temporal clustering bug fix

GiulioRossetti · Feb 17, 2021 · 325d816 · 325d816
1 parent 40a4ad8
commit 325d816
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/cdlib/classes/temporal_clustering.py b/cdlib/classes/temporal_clustering.py
@@ -217,7 +217,12 @@ def lifecycle_polytree(self, method=None, two_sided=False):
             lifecycle = self.community_matching(method, two_sided)
 
         pt = nx.DiGraph()
-        for u, v, w in lifecycle:
-            pt.add_edge(u, v, weight=w)
+        if len(lifecycle[0]) == 3:
+            for u, v, w in lifecycle:
+                pt.add_edge(u, v, weight=w)
+        else:
+            # implicit matching
+            for u, v in lifecycle:
+                pt.add_edge(u, v)
 
         return pt