I have a flow where the vizualization is adding a bunch of List->List->Dict boxes which makes it hard to read. How can I remove those?

standardizer_task = submit_standardizer_job(files, bucket, 'mcule', version)
standardizer_wait_task = AWSClientWait(client='batch', waiter_name='JobComplete', max_retries=2, retry_delay=datetime.timedelta(minutes=1))(waiter_kwargs={'jobs': [standardizer_task]})

Hey @Trevor Kramer, I’m not sure you can get rid of these with the visualize method currently, but you can package your data manipulation into a task to keep Prefect from visualizing autogenerated tasks by adding something like:

@task
def prepare_waiter_kwargs(standardizer_task):
    return {'jobs': [standardizer_task]}

to remove some of those boxes from the graphviz pdf.

thanks - that helped