Forgot to self env

pydyton/communities/kclique.py

@@ -2,6 +2,8 @@ import networkx as nx
 
 from .base import EpochCommunity
 
+# TODO: this is probably bugged, performs terribly for bubble
+
 class KCliqueCommunity(EpochCommunity):
     def __init__(self, k=3, threshold=300, epoch=604800, **kwargs):
         super().__init__(epoch, **kwargs)
@@ -11,7 +13,7 @@ class KCliqueCommunity(EpochCommunity):
     def process(self, network):
         while True:
             yield self.tick()
-            g = self.next_epoch(env.now)
+            g = self.next_epoch(self.env.now)
 
             G = nx.Graph()
             G.add_nodes_from(g.nodes())

pydyton/communities/louvain.py

+from collections import defaultdict
+
 from community import best_partition as louvain_partition
 
 from .base import EpochCommunity
@@ -11,7 +13,7 @@ class LouvainCommunity(EpochCommunity):
     def process(self, network):
         while True:
             yield self.tick()
-            g = self.next_epoch(env.now)
+            g = self.next_epoch(self.env.now)
 
             p = louvain_partition(g, weight='duration')
             communities = defaultdict(set)