Allow to control the width of the heatmap plot

[ravwojdyla]

Reason: default size might be a bit too small for a larger datasets.

[codecov]

Codecov Report

Merging #2242 (6995c03) into master (b6e90c8) will decrease coverage by 0.00%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##           master    #2242      +/-   ##
==========================================
- Coverage   51.61%   51.61%   -0.01%     
==========================================
  Files          91       91              
  Lines       13147    13144       -3     
==========================================
- Hits         6786     6784       -2     
+ Misses       6361     6360       -1     

Impacted Files	Coverage Δ
shap/plots/_heatmap.py	94.80% <100.00%> (-0.07%)	⬇️
shap/_explanation.py	54.67% <0.00%> (-0.20%)	⬇️
shap/models/_teacher_forcing.py	79.88% <0.00%> (+0.45%)	⬆️
shap/models/_text_generation.py	73.73% <0.00%> (+0.73%)	⬆️

📣 Codecov can now indicate which changes are the most critical in Pull Requests. Learn more

[slundberg]

Excellent, thank you! Sorry I have been out for a while and let this sit. Will merge once the new CI tests pass.