flyte ❤️ Hi 👋 , hope someone can help me with a conditional workflow that I am struggling with. To be clear, the workflow runs as I expected, but the graph that is produced is misleading. This is the workflow that I am running:

@workflow
def some_training_job(n_training_jobs: int = 1):
    training, validation, test = do_train_fech_data()
    image_name = build_and_push_image()
    ( conditional("Single train or HPO")
        .if_(n_training_jobs > 1)
        .then(run_hyperparameters_optimisation_job(
            single_training_job_image=image_name,
            training=training, 
            validation=validation, 
            test=test,
            n_training_jobs=n_training_jobs
        ))
        .else_()
        .then(run_single_trainin_job(
            single_training_job_image=image_name,
            training=training, 
            validation=validation, 
            test=test
        ))
    )

The workflow works fine, it fetches the data, builds an image and after evaluating

n_training_jobs

it runs

run_hyperparameters_optimisation_job

or

run_single_trainin_job

. However, the graph that it produced looks like the data and the image outputs are not used for the conditional box, which is not what it is happening. Do you think this is a bug with conditionals and the ui? or am I constructing the workflow in the wrong way (even though it runs correctly)

Looks like a UI bug to me. Can you create an issue? https://github.com/flyteorg/flyte/issues/new?assignees=&labels=enhancement%2C+untr[…]jects=&template=ui_feature_request.yaml&title=%5BUI+Feature%5D+

Thanks for looking into it 🙇 I will create an issue 👍