Skip to content

Commit

Permalink
Adds node name to hamilton dask DAG
Browse files Browse the repository at this point in the history
Fixes #161
  • Loading branch information
elijahbenizzy committed May 4, 2023
1 parent 3651a26 commit d3acc43
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion hamilton/experimental/h_dask.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ def execute_node(self, node: node.Node, kwargs: typing.Dict[str, typing.Any]) ->
:param kwargs: the arguments that should be passed to it.
:return: returns a dask delayed object.
"""
return delayed(node.callable)(**kwargs)
return delayed(node.callable, name=node.name)(**kwargs)

def build_result(self, **outputs: typing.Dict[str, typing.Any]) -> typing.Any:
"""Builds the result and brings it back to this running process.
Expand Down

0 comments on commit d3acc43

Please sign in to comment.