Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add unused node check for compile_dag_node #49382

Open
wants to merge 13 commits into
base: master
Choose a base branch
from
Prev Previous commit
Next Next commit
fix no attribute error
Signed-off-by: Lydia <wenqix3@uw.edu>
  • Loading branch information
LydiaXwQ committed Dec 20, 2024
commit 018200488cd32b4e7287df0b158d7012e9f351f1
5 changes: 2 additions & 3 deletions python/ray/dag/compiled_dag_node.py
Original file line number Diff line number Diff line change
Expand Up @@ -994,13 +994,12 @@ def _preprocess(self) -> None:
if isinstance(task.dag_node, InputNode):
assert self.input_task_idx is None, "More than one InputNode found"
self.input_task_idx = idx
elif isinstance(task.dag_node, InputAttributeNode):
self.input_attr_task_idxs.append(idx)

# handle_unused_attributes:
# Save input attributes in a set.
input_node = task.dag_node
input_attributes.update(input_node.input_attribute_nodes.keys())
elif isinstance(task.dag_node, InputAttributeNode):
self.input_attr_task_idxs.append(idx)

# Find the (multi-)output node to the DAG.
for idx, task in self.idx_to_task.items():
Expand Down
Loading