Make plot without grid nodes

[robertdj]

Hi!

I am a recent adopter of Ploomber and enjoying it. Thanks a lot for making this framework.

I am using grid in my pipeline.yaml and it works great. The only downside is that when visualizing the DAG with pipeline plot all nodes are included (and if there are enough nodes the html show nothing).

However, since nodes made by a grid step are related I would like them to appear as a single node in the DAG. Would it be possible to make an option for pipeline plot such that the user can choose whether or not to collapse grid nodes?

[idomic]

Thanks for reporting this @robertdj, sounds like a parameter to pass to the plotting function.
Would you like to submit a pull request?

[robertdj]

Thanks for the opportunity. I will give it a try.

[edublancas]

if you need help, feel free to ping us here or in Slack! https://ploomber.io/community/