https://flyte.org logo
Join Slack
Powered by
Hey! Iā€™m using `with_overrides(_node_name_=f"{vari...
# flyte-support
s
Hey! Iā€™m using 
with_overrides(_node_name_=f"{variable}-some-task")
to set a less cryptic name for my dynamic task (to have 
{variable}-some-task
show in the UI instead of 
d1-n-dn0
), and that works on first glance, but they get changed when the workload starts running; and stay that way after completion. Is this a bug or expected behaviour? Any way to get the name to stay?
t
have you noticed this on non-dynamic tasks?
s
I can check šŸ‘
For non-dynamic tasks it seems to work. Unfortunately, dynamic tasks is actually where this is needed, because Im using that to generate a bunch of tasks for which it would be nice to be able to differentiate them in the UI šŸ˜…
t
@late-eye-50215 can you take a look at this? @shy-evening-51366 can you download the 
futures.pb
file (it should be in the same place where you normally find the 
outputs.pb
) for the dynamic task?
also could you copy/paste the code for the dynamic task? (feel free to redact of course)
s
Hmm the 
futures.pb
contains some sensitive credentials, let me make a version that can be shared šŸ‘
Okay, took me a bit of time to repro and get down to a minimal example. If you have just 1 task created in the loop, I did not observe the issue. When I add another (
task_some_other_task
in this example), the issue occurs.
Copy code
from flytekit import Resources, WorkflowFailurePolicy, dynamic, task, workflow


@task(
    limits=Resources(mem="500Mi"),
)
def some_task(
    val: str,
):
    print(val)


@dynamic(limits=Resources(mem="500Mi"))
def base_workflow():
    # set up task promises
    for val in ["one", "two", "three"]:
        task_some_task = some_task(
            val=val,
        ).with_overrides(node_name=f"{val}-some-task")
        task_some_other_task = some_task(
            val=val,
        ).with_overrides(node_name=f"{val}-some-other-task")

        task_some_task >> task_some_other_task


@workflow(    failure_policy=WorkflowFailurePolicy.FAIL_AFTER_EXECUTABLE_NODES_COMPLETE,
)
def wf():
    base_workflow()
Starts with the correct overridden names for all tasks, but after the first task 
task_some_task
completes, the names of subsequent tasks get updated to random string:
futures.pb file
futures(1).pb
šŸŽ‰ 1
66 Views
Open in Slack
PreviousNext
Powered by