Change default heatmap width

polakowo · Feb 22, 2021 · 6c555ac · 6c555ac
1 parent 4152f22
commit 6c555ac
Show file tree

Hide file tree

Showing 7 changed files with 2 additions and 2 deletions.
diff --git a/docs/img/heatmap.png b/docs/img/heatmap.png
diff --git a/docs/img/sr_heatmap.png b/docs/img/sr_heatmap.png
diff --git a/docs/img/sr_heatmap_slider.gif b/docs/img/sr_heatmap_slider.gif
diff --git a/docs/img/sr_overlay_with_heatmap.png b/docs/img/sr_overlay_with_heatmap.png
diff --git a/static/local_extrema.png b/static/local_extrema.png
diff --git a/vectorbt/generic/accessors.py b/vectorbt/generic/accessors.py
@@ -1134,7 +1134,7 @@ def overlay_with_heatmap(self, other, trace_kwargs=None, heatmap_kwargs=None,
         if fig is None:
             fig = make_subplots(specs=[[{"secondary_y": True}]])
             if 'width' in layout:
-                fig.update_layout(width=layout['width'] + 150)
+                fig.update_layout(width=layout['width'] + 100)
         fig.update_layout(**layout_kwargs)
 
         other.vbt.ts_heatmap(**heatmap_kwargs, add_trace_kwargs=add_trace_kwargs, fig=fig)

diff --git a/vectorbt/generic/plotting.py b/vectorbt/generic/plotting.py
@@ -607,7 +607,7 @@ def __init__(self, data=None, x_labels=None, y_labels=None,trace_kwargs=None,
                     [0, 1],
                     [0.3 * max_width, max_width]
                 ))
-                width = min(width + 150, max_width)  # account for colorbar
+                width = min(width + 100, max_width)  # account for colorbar
                 height = math.ceil(renormalize(
                     y_len / (x_len + y_len),
                     [0, 1],