Skip to content

Commit e6b53f6

Browse files
committed
Fix typo: afftected
1 parent 32ea951 commit e6b53f6

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

mars/optimization/logical/core.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)