Feature request: Option to ignore imputation in the `pm.model_to_graphviz` plate model
Open
Nobody has claimed this yet.
enhancements
- Dominant language
- Python
- Stars
- 9.8k
- Forks
- 2.3k
- Avg merge
- 21h 39m
- Merged PRs (30d)
- 5
Description
Description of your problem
Using the imputation feature leads to an automatic split-up of the model variables.
In the pm.model_to_graphviz visualization that causes arrows to go in the reverse direction:

Versions and main components
- PyMC3 Version:
main
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start at the pm.model_to_graphviz entry point and reproduce the imputation example described in the issue. Determine how an option to ignore imputation should affect the rendered model graph, then verify that the split variables and reversed arrows no longer appear when the option is used.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- data-visualization
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100