Fix plot_lm handling of multidimensional data with y_model (#1992) #2408
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
Fix plot_lm functionality when using multidimensional data with y_model parameter. Previously, attempting to use plot_lm with both multidimensional y and y_model would result in dimension mismatch errors.
Also adds a test to check for the proper functioning of plot_lm when using multidimensional data with y_model parameter.
Addresses issue #1992.
Checklist
Following problematic cases in this notebook now work: https://gist.github.com/colehaus/b2aa195687f53ef7c86eec2ebb3798c2
📚 Documentation preview 📚: https://arviz--2408.org.readthedocs.build/en/2408/