Update: specify plot size

Author: heptaflar

kaleidoscope/interface/plot.py

@@ -26,6 +26,7 @@ def plot(
         ylim: tuple[Any, Any] | None = None,
         title: str | None = None,
         fn: str | None = None,
+        plot_size: tuple[Any, Any] | None = None,
         show: bool = False,
         **kwargs,
     ) -> Figure:
@@ -39,6 +40,7 @@ def plot(
         :param ylim: The limits for the y-axis.
         :param title: The title of the plot.
         :param fn: The file name to save the figure.
+        :param plot_size: The size of the plot.
         :param show: Show the figure.
         :param kwargs: Additional keyword arguments.
         :return: The plot.

kaleidoscope/val/plots.py

@@ -10,6 +10,7 @@
 
 import dask.array as da
 import numpy as np
+from IPython.core.pylabtools import figsize
 from matplotlib import colors as plc
 from matplotlib import pyplot as plt
 from matplotlib.axes import Axes
@@ -35,6 +36,7 @@ def plot(
         ylim: tuple[Any, Any] | None = None,
         title: str | None = None,
         fn: str | None = None,
+        plot_size: tuple[Any, Any] | None = None,
         show: bool = False,
         *,
         cbar_label: str | None = None,
@@ -51,6 +53,7 @@ def plot(
         if projection is None:
             projection = self.projection
         fig, ax = plt.subplots(
+            figsize=plot_size,
             subplot_kw={"projection": projection},
         )
         cbar_kwargs = {}
@@ -130,14 +133,15 @@ def plot(
         ylim: tuple[Any, Any] | None = None,
         title: str | None = None,
         fn: str | None = None,
+        plot_size: tuple[Any, Any] | None = None,
         show: bool = False,
         *,
         bins: int | None = None,
         density: bool = False,
         log: bool = False,
         hist_range: tuple[Any, Any] | None = None,
     ) -> Figure:
-        fig, ax = plt.subplots()
+        fig, ax = plt.subplots(figsize=plot_size)
         data.plot.hist(
             ax=ax, bins=bins, range=hist_range, density=density, log=log
         )
@@ -166,6 +170,7 @@ def plot(
         ylim: tuple[Any, Any] | None = None,
         title: str | None = None,
         fn: str | None = None,
+        plot_size: tuple[Any, Any] | None = None,
         show: bool = False,
         *,
         bins: tuple[int, int] | None = None,
@@ -177,7 +182,7 @@ def plot(
         vmin: Any | None = None,
         vmax: Any | None = None,
     ) -> Figure:
-        fig, ax = plt.subplots()
+        fig, ax = plt.subplots(figsize=plot_size)
         cbar_kwargs = {}
         if cbar_label is not None:
             cbar_kwargs["label"] = cbar_label
@@ -217,14 +222,15 @@ def plot(
         ylim: tuple[Any, Any] | None = None,
         title: str | None = None,
         fn: str | None = None,
+        plot_size: tuple[Any, Any] | None = None,
         show: bool = False,
         *,
         point_alpha: Any | None = None,
         point_color: str | None = None,
         point_marker: str | None = ".",
         sample_count: int = 4000,
     ) -> Figure:
-        fig, ax = plt.subplots()
+        fig, ax = plt.subplots(figsize=plot_size)
         rand(data, sample_count).plot.scatter(
             ax=ax,
             x="x",
@@ -262,12 +268,13 @@ def plot(
         ylim: tuple | None = None,
         title: str | None = None,
         fn: str | None = None,
+        plot_size: tuple[Any, Any] | None = None,
         show: bool = False,
         *,
         point_marker: str = ".",
         group_by: str | None = "time.month",
     ) -> Figure:
-        fig, ax = plt.subplots()
+        fig, ax = plt.subplots(figsize=plot_size)
 
         if group_by is not None:
             for _, period in time_series(data).groupby(group_by):