Merge pull request #159 from inferno-pytorch/nx2

Change networkx API.

Author: constantinpape

.travis.yml

@@ -19,7 +19,7 @@ install:
   - conda update -q conda
   - conda create -q -n test-environment python=$TRAVIS_PYTHON_VERSION
   - source activate test-environment
-  - conda install -c conda-forge networkx=1.11 h5py scikit-image pyyaml dill tensorboardx
+  - conda install -c conda-forge networkx h5py scikit-image pyyaml dill tensorboardx
   - conda install -c pytorch $PYTORCH_CONDA
   - conda install -c $TORCHVISION_CHANNEL $TORCHVISION_CONDA
 

inferno/extensions/containers/graph.py

@@ -386,7 +386,8 @@ def get_parameters_for_nodes(self, names, named=False):
 
     def clear_payloads(self, graph=None):
         graph = self.graph if graph is None else graph
-        for source, target in graph.edges_iter():
+        for edge in list(graph.edges(data=True)):
+            source, target, _ = edge
             if 'payload' in graph[source][target]:
                 del graph[source][target]['payload']