Adapt to new networkX API

inferno-pytorch · Feb 15, 2020 · 4c732db · 4c732db
1 parent 4ebc5e8
commit 4c732db
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/inferno/extensions/containers/graph.py b/inferno/extensions/containers/graph.py
@@ -130,7 +130,7 @@ def is_node_in_graph(self, name):
         -------
         bool
         """
-        return name in self.graph.node
+        return name in self.graph.nodes
 
     def is_source_node(self, name):
         """
@@ -187,7 +187,7 @@ def output_nodes(self):
         list
             A list of names (str) of the output nodes.
         """
-        return [name for name, node_attributes in self.graph.node.items()
+        return [name for name, node_attributes in self.graph.nodes.items()
                 if node_attributes.get('is_output_node', False)]
 
     @property
@@ -201,7 +201,7 @@ def input_nodes(self):
         list
             A list of names (str) of the input nodes.
         """
-        return [name for name, node_attributes in self.graph.node.items()
+        return [name for name, node_attributes in self.graph.nodes.items()
                 if node_attributes.get('is_input_node', False)]
 
     @property