@@ -157,7 +157,7 @@ def _replace_subgraph(
157157 If the input key of the removed node's successor can't be found in the subgraph.
158158 Or some of the nodes of the subgraph are in removed ones.
159159 """
160- infected_successors = set ()
160+ affected_successors = set ()
161161
162162 output_to_node = dict ()
163163 removed_nodes = removed_nodes or set ()
@@ -170,15 +170,15 @@ def _replace_subgraph(
170170 output_to_node [output .key ] = node
171171
172172 for node in removed_nodes :
173- for infected_successor in self ._graph .iter_successors (node ):
174- if infected_successor not in removed_nodes :
175- infected_successors .add (infected_successor )
176- # Check whether infected successors' inputs are in subgraph
177- for infected_successor in infected_successors :
178- for inp in infected_successor .inputs :
173+ for affected_successor in self ._graph .iter_successors (node ):
174+ if affected_successor not in removed_nodes :
175+ affected_successors .add (affected_successor )
176+ # Check whether affected successors' inputs are in subgraph
177+ for affected_successor in affected_successors :
178+ for inp in affected_successor .inputs :
179179 if inp .key not in output_to_node :
180180 raise ValueError (
181- f"The output { inp } of node { infected_successor } is missing in the subgraph"
181+ f"The output { inp } of node { affected_successor } is missing in the subgraph"
182182 )
183183 for node in removed_nodes :
184184 self ._graph .remove_node (node )
@@ -194,7 +194,7 @@ def _replace_subgraph(
194194 for node in graph .iter_nodes ():
195195 self ._graph .add_node (node )
196196
197- for node in itertools .chain (graph .iter_nodes (), infected_successors ):
197+ for node in itertools .chain (graph .iter_nodes (), affected_successors ):
198198 for inp in node .inputs :
199199 pred_node = output_to_node [inp .key ]
200200 self ._graph .add_edge (pred_node , node )
0 commit comments