bugfix

GiulioRossetti · Aug 9, 2017 · bbc7d21 · bbc7d21
1 parent f0f7dac
commit bbc7d21
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ndlib/viz/mpl/ComparisonViz.py b/ndlib/viz/mpl/ComparisonViz.py
@@ -74,7 +74,7 @@ def plot(self, filename, percentile=90):
 
         mx = 0
         i, h = 0, 0
-        for k, l in pres.iteritems():
+        for k, l in future.utils.iteritems(pres):
             j = 0
             for st in l:
                 mx = len(l[st][0])

diff --git a/ndlib/viz/mpl/DiffusionViz.py b/ndlib/viz/mpl/DiffusionViz.py
@@ -49,7 +49,7 @@ def plot(self, filename=None, percentile=90):
 
         mx = 0
         i = 0
-        for k, l in pres.iteritems():
+        for k, l in future.utils.iteritems(pres):
             mx = len(l[0])
             if self.normalized:
                 plt.plot(range(0, mx), l[1]/self.nnodes, lw=2, label=self.srev[k], alpha=0.5, color=cols[i])