plothist 1.6.0__py3-none-any.whl → 1.8.0__py3-none-any.whl

plothist/__init__.py

@@ -9,12 +9,7 @@ from .comparison import (
     get_ratio,
     get_ratio_variances,
 )
-from .histogramming import (
-    create_axis,
-    flatten_2d_hist,
-    make_2d_hist,
-    make_hist,
-)
+from .histogramming import create_axis, flatten_2d_hist, make_2d_hist, make_hist
 from .plothist_style import (
     add_luminosity,
     add_text,
@@ -43,6 +38,7 @@ from .variable_registry import (
     get_variable_from_registry,
     remove_variable_registry_parameters,
     update_variable_registry,
+    update_variable_registry_binning,
     update_variable_registry_ranges,
 )
 
@@ -81,6 +77,7 @@ __all__ = [
     "set_fitting_ylabel_fontsize",
     "set_style",
     "update_variable_registry",
+    "update_variable_registry_binning",
     "update_variable_registry_ranges",
 ]
 

plothist/_version.py

@@ -1,7 +1,14 @@
 # file generated by setuptools-scm
 # don't change, don't track in version control
 
-__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
 
 TYPE_CHECKING = False
 if TYPE_CHECKING:
@@ -9,13 +16,19 @@ if TYPE_CHECKING:
     from typing import Union
 
     VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
 else:
     VERSION_TUPLE = object
+    COMMIT_ID = object
 
 version: str
 __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
 
-__version__ = version = '1.6.0'
-__version_tuple__ = version_tuple = (1, 6, 0)
+__version__ = version = '1.8.0'
+__version_tuple__ = version_tuple = (1, 8, 0)
+
+__commit_id__ = commit_id = None

plothist/comparison.py

@@ -49,7 +49,7 @@ def _is_unweighted(hist: bh.Histogram) -> bool:
     bool
         True if the histogram is unweighted, False otherwise.
     """
-    return np.allclose(hist.variances(), hist.values(), equal_nan=True)
+    return bool(np.allclose(hist.variances(), hist.values(), equal_nan=True))
 
 
 def get_asymmetrical_uncertainties(

plothist/examples/2d_hist/2d_hist_correlations.py

@@ -20,7 +20,7 @@ from plothist import (
     get_variable_from_registry,
     make_2d_hist,
     plot_2d_hist,
-    update_variable_registry_ranges,
+    update_variable_registry_binning,
 )
 
 # No need to redo this step if the registry was already created before
@@ -28,7 +28,7 @@ variable_keys = ["variable_0", "variable_1", "variable_2"]
 unique_id = str(int(time.time() * 1000))[-8:]  # unique ID based on current time
 temporary_registry_path = f"./_temporary_variable_registry_{unique_id}.yaml"
 create_variable_registry(variable_keys, path=temporary_registry_path)
-update_variable_registry_ranges(df, variable_keys, path=temporary_registry_path)
+update_variable_registry_binning(df, variable_keys, path=temporary_registry_path)
 
 # Get all the correlation plot between the variables
 variable_keys_combinations = list(combinations(variable_keys, 2))

plothist/examples/advanced/1d_side_by_side_with_numbers.py

@@ -42,7 +42,7 @@ plot_hist(histos, ax=ax, label=labels, color=colors)
 
 # Add the number of entries on top of each bar
 # Get the correct shift in x-axis for each bar
-def calculate_shifts(width, n_bars):
+def calculate_shifts(width: float, n_bars: int) -> np.ndarray:
     half_width = width / 2
     shift = np.linspace(-half_width, half_width, n_bars, endpoint=False)
     shift += width / (2 * n_bars)

plothist/examples/advanced/asymmetry_comparison_advanced.py

@@ -11,6 +11,7 @@ from plothist_utils import get_dummy_data
 df = get_dummy_data()
 
 ###
+import numpy as np
 from scipy.stats import norm
 
 from plothist import (
@@ -25,11 +26,11 @@ from plothist import (
 
 
 # Define some random functions that will be used as Data fit functions
-def f1(x):
+def f1(x: np.ndarray) -> np.ndarray:
     return 4000 * norm.pdf(x, loc=-0.5, scale=1.6)
 
 
-def f2(x):
+def f2(x: np.ndarray) -> np.ndarray:
     return 4000 * norm.pdf(x, loc=0.5, scale=1.6)
 
 

plothist/examples/utility/color_palette_squares.py

@@ -13,16 +13,21 @@ from matplotlib import patches
 from plothist import get_color_palette
 
 
-def create_palette_plot(colors, fig_name, add_text=False, add_black_border=False):
+def create_palette_plot(
+    colors: list[str] | list[tuple[float, float, float]],
+    fig_name: str,
+    add_text: bool = False,
+    add_black_border: bool = False,
+):
     ncolors = len(colors)
 
     # Create a figure and axis
     fig, ax = plt.subplots(figsize=(ncolors, 1))
 
     # Plot the colored squares with small spacing
-    square_size = 1
+    square_size = 1.0
     spacing = 0.1
-    x = 0
+    x = 0.0
 
     for color in colors:
         rect = patches.Rectangle((x, 0), square_size, square_size, color=color)

plothist/histogramming.py

@@ -1,8 +1,7 @@
 from __future__ import annotations
 
 import warnings
-from collections.abc import Sequence
-from typing import Callable
+from collections.abc import Callable, Sequence
 
 import boost_histogram as bh
 import numpy as np
@@ -109,6 +108,7 @@ def make_hist(
     bins: int | list[float] | np.ndarray = 50,
     range: tuple[float | str, float | str] | None = None,
     weights: float | list[float] | np.ndarray = 1,
+    mute_warning: bool = False,
 ) -> bh.Histogram:
     """
     Create a histogram object and fill it with the provided data.
@@ -129,6 +129,8 @@ def make_hist(
         Weight(s) to apply to the data points (default is 1).
         If a float, a single weight is applied to all data points.
         If an array-like, weights are applied element-wise.
+    mute_warning : bool, optional
+        Whether to mute warnings about data outside the binning range (default is False).
 
     Returns
     -------
@@ -160,7 +162,7 @@ def make_hist(
         range_coverage = h.sum().value / n_data
 
         # Issue a warning if more than 1% of the data is outside of the binning range
-        if range_coverage < 0.99:
+        if range_coverage < 0.99 and not mute_warning:
             warnings.warn(
                 f"Only {100 * range_coverage:.2f}% of data contained in the binning range [{axis.edges[0]}, {axis.edges[-1]}].",
                 category=RangeWarning,
@@ -177,6 +179,7 @@ def make_2d_hist(
         tuple[float | str, float | str] | None, tuple[float | str, float | str] | None
     ] = (None, None),
     weights: float | list[float] | np.ndarray = 1,
+    mute_warning: bool = False,
 ) -> bh.Histogram:
     """
     Create a 2D histogram object and fill it with the provided data.
@@ -198,6 +201,8 @@ def make_2d_hist(
         Weight(s) to apply to the data points (default is 1).
         If a float, a single weight is applied to all data points.
         If an array-like, weights are applied element-wise.
+    mute_warning : bool, optional
+        Whether to mute warnings about data outside the binning range (default is False).
 
     Returns
     -------
@@ -245,7 +250,7 @@ def make_2d_hist(
         range_coverage = h.sum().value / n_data
 
         # Issue a warning if more than 1% of the data is outside of the binning range
-        if range_coverage < 0.99:
+        if range_coverage < 0.99 and not mute_warning:
             warnings.warn(
                 f"Only {100 * range_coverage:.2f}% of data contained in the binning range ([{x_axis.edges[0]}, {x_axis.edges[-1]}], [{y_axis.edges[0]}, {y_axis.edges[-1]}]).",
                 category=RangeWarning,

plothist/plothist_style.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from collections.abc import Callable
+
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 import numpy as np
@@ -82,9 +84,9 @@ def cubehelix_palette(
     p. 289-295.
     """
 
-    def f(x0, x1):
+    def f(x0: float, x1: float) -> Callable[[float], np.ndarray]:
         # Adapted from matplotlib
-        def color(lambda_):
+        def color(lambda_: float) -> np.ndarray:
             # emphasise either low intensity values (gamma < 1),
             # or high intensity values (gamma > 1)
             lambda_gamma = lambda_**gamma
@@ -187,27 +189,36 @@ def set_fitting_ylabel_fontsize(ax: plt.Axes) -> float:
     float
         The adjusted font size for the ylabel text.
     """
-    ylabel_fontsize = ax.yaxis.get_label().get_fontsize()
+    ylabel_fontsize = float(ax.yaxis.get_label().get_fontsize())
 
     # Force renderer to be initialized
     ax.figure.canvas.draw()
 
-    while (
+    current_extent = (
         ax.yaxis.get_label()
         .get_window_extent(renderer=ax.figure.canvas.get_renderer())
         .transformed(ax.transData.inverted())
-        .y1
-        > ax.get_ylim()[1]
-    ):
+    )
+
+    y_alignment = plt.rcParams["yaxis.labellocation"]
+
+    while (
+        y_alignment in ["center", "bottom"] and current_extent.y1 > ax.get_ylim()[1]
+    ) or (y_alignment == "top" and current_extent.y0 < ax.get_ylim()[0]):
         ylabel_fontsize -= 0.1
 
         if ylabel_fontsize <= 0:
-            raise ValueError(
-                "Only a y-label with a negative font size would fit on the y-axis."
-            )
+            msg = "Only a y-label with a negative font size would fit on the y-axis."
+            raise ValueError(msg)
 
         ax.get_yaxis().get_label().set_size(ylabel_fontsize)
 
+        current_extent = (
+            ax.yaxis.get_label()
+            .get_window_extent(renderer=ax.figure.canvas.get_renderer())
+            .transformed(ax.transData.inverted())
+        )
+
     return ylabel_fontsize
 
 
@@ -237,7 +248,7 @@ def add_text(
         Draw a white rectangle under the text, by default False.
     ax : matplotlib.axes.Axes, optional
         Figure axis, by default None.
-    kwargs : dict
+    kwargs : dict[str, Any]
         Keyword arguments to be passed to the ax.text() function.
         In particular, the keyword arguments ha and va, which are set by default to accommodate to the x and y aliases, can be used to change the text alignment.
 
@@ -341,7 +352,7 @@ def add_luminosity(
         Draw a white rectangle under the text, by default False.
     ax : matplotlib.axes.Axes, optional
         Figure axis, by default None.
-    kwargs : dict
+    kwargs : dict[str, Any]
         Keyword arguments to be passed to the ax.text() function.
         In particular, the keyword arguments ha and va, which are set to "left" (or "right" if x="right") and "bottom" by default, can be used to change the text alignment.
 
@@ -393,7 +404,7 @@ def plot_reordered_legend(ax: plt.Axes, order: list[int], **kwargs) -> None:
     order : list[int]
         A list of integers specifying the new order of the legend items.
         The integers refer to the indices of the current legend items.
-    kwargs : dict, optional
+    kwargs : dict[str, Any], optional
         Keyword arguments to be passed to the ax.legend() function, such as the legend location (loc).
 
     Returns

plothist/plotters.py

@@ -5,7 +5,8 @@ Collection of functions to plot histograms
 from __future__ import annotations
 
 import re
-from typing import Callable
+from collections.abc import Callable
+from typing import Any
 
 import boost_histogram as bh
 import matplotlib.pyplot as plt
@@ -23,9 +24,9 @@ from plothist.plothist_style import set_fitting_ylabel_fontsize
 
 
 def create_comparison_figure(
-    figsize: tuple[float, float] = (6, 5),
+    figsize: tuple[float, float] | None = (6, 5),
     nrows: int = 2,
-    gridspec_kw: dict | None = None,
+    gridspec_kw: dict[str, Any] | None = None,
     hspace: float = 0.15,
 ) -> tuple[plt.Figure, np.ndarray]:
     """
@@ -35,9 +36,10 @@ def create_comparison_figure(
     ----------
     figsize : tuple[float, float], optional
         Figure size in inches. Default is (6, 5).
+        If None is provided, the default matplotlib figure size from rcParams["figure.figsize"] is used.
     nrows : int, optional
         Number of rows in the subplot grid. Default is 2.
-    gridspec_kw : dict | None, optional
+    gridspec_kw : dict[str, Any] | None, optional
         Additional keyword arguments for the GridSpec. Default is None.
         If None is provided, this is set to {"height_ratios": [4, 1]}.
     hspace : float, optional
@@ -104,8 +106,8 @@ def plot_2d_hist(
     fig: plt.Figure | None = None,
     ax: plt.Axes | None = None,
     ax_colorbar: plt.Axes | None = None,
-    pcolormesh_kwargs: dict | None = None,
-    colorbar_kwargs: dict | None = None,
+    pcolormesh_kwargs: dict[str, Any] | None = None,
+    colorbar_kwargs: dict[str, Any] | None = None,
     square_ax: bool = True,
 ) -> tuple[plt.Figure, plt.Axes, plt.Axes]:
     """
@@ -121,9 +123,9 @@ def plot_2d_hist(
         The Axes instance for plotting. If fig, ax and ax_colorbar are all None, a new figure will be created. Default is None.
     ax_colorbar : matplotlib.axes.Axes | None, optional
         The Axes instance for the colorbar. If fig, ax and ax_colorbar are all None, a new figure will be created. Default is None.
-    pcolormesh_kwargs : dict | None, optional
+    pcolormesh_kwargs : dict[str, Any] | None, optional
         Additional keyword arguments forwarded to ax.pcolormesh(). Default is None.
-    colorbar_kwargs : dict | None, optional
+    colorbar_kwargs : dict[str, Any] | None, optional
         Additional keyword arguments forwarded to ax.get_figure().colorbar(). Default is None.
     square_ax : bool, optional
         Whether to make the main ax square. Default is True.
@@ -242,9 +244,9 @@ def plot_2d_hist_with_projections(
     xlabel_y_projection: str | None = None,
     colorbar_label: str | None = None,
     offset_x_labels: bool = False,
-    pcolormesh_kwargs: dict | None = None,
-    colorbar_kwargs: dict | None = None,
-    plot_hist_kwargs: dict | None = None,
+    pcolormesh_kwargs: dict[str, Any] | None = None,
+    colorbar_kwargs: dict[str, Any] | None = None,
+    plot_hist_kwargs: dict[str, Any] | None = None,
     figsize: tuple[float, float] = (6, 6),
 ) -> tuple[plt.Figure, plt.Axes, plt.Axes, plt.Axes, plt.Axes]:
     """
@@ -266,11 +268,11 @@ def plot_2d_hist_with_projections(
         Label for the colorbar. Default is None.
     offset_x_labels : bool, optional
         Whether to offset the x labels to avoid overlapping with the exponent label (i.e. "10^X") of the axis. Default is False.
-    pcolormesh_kwargs : dict | None, optional
+    pcolormesh_kwargs : dict[str, Any] | None, optional
         Keyword arguments for the pcolormesh call. Default is None.
-    colorbar_kwargs : dict | None, optional
+    colorbar_kwargs : dict[str, Any] | None, optional
         Keyword arguments for the colorbar call. Default is None.
-    plot_hist_kwargs : dict | None, optional
+    plot_hist_kwargs : dict[str, Any] | None, optional
         Keyword arguments for the plot_hist call (x and y projections). Default is None.
     figsize : tuple[float, float], optional
         Figure size in inches. Default is (6, 6). To get square bins if the figure is not square shaped, be sure to set the bins and the ranges of the histogram according to the ratio of the figure width and height.
@@ -767,7 +769,7 @@ def _get_math_text(text: str) -> str:
     return text
 
 
-def _get_model_type(components: list) -> str:
+def _get_model_type(components: list[bh.Histogram | Callable[[Any], Any]]) -> str:
     """
     Check that all components of a model are either all histograms or all functions
     and return the type of the model components.
@@ -805,9 +807,9 @@ def plot_model(
     ) = None,
     xlabel: str | None = None,
     ylabel: str | None = None,
-    stacked_kwargs: dict | None = None,
-    unstacked_kwargs_list: list[dict] | None = None,
-    model_sum_kwargs: dict | None = None,
+    stacked_kwargs: dict[str, Any] | None = None,
+    unstacked_kwargs_list: list[dict[str, Any]] | None = None,
+    model_sum_kwargs: dict[str, Any] | None = None,
     function_range: tuple[float, float] | None = None,
     model_uncertainty: bool = True,
     model_uncertainty_label: str = "Model stat. unc.",
@@ -835,11 +837,11 @@ def plot_model(
         The label for the x-axis. Default is None.
     ylabel : str | None, optional
         The label for the y-axis. Default is None.
-    stacked_kwargs : dict | None, optional
+    stacked_kwargs : dict[str, Any] | None, optional
         The keyword arguments used when plotting the stacked components in plot_hist() or plot_function(), one of which is called only once. Default is None.
-    unstacked_kwargs_list : list[dict] | None, optional
+    unstacked_kwargs_list : list[dict[str, Any]] | None, optional
         The list of keyword arguments used when plotting the unstacked components in plot_hist() or plot_function(), one of which is called once for each unstacked component. Default is None.
-    model_sum_kwargs : dict | None, optional
+    model_sum_kwargs : dict[str, Any] | None, optional
         The keyword arguments for the plot_hist() function for the sum of the model components.
         Has no effect if all the model components are stacked or if the model is one unstacked element.
         The special keyword "show" can be used with a boolean to specify whether to show or not the sum of the model components.
@@ -947,6 +949,7 @@ def plot_model(
             unstacked_colors,
             unstacked_labels,
             unstacked_kwargs_list,
+            strict=True,
         ):
             if model_type == "histograms":
                 unstacked_kwargs.setdefault("histtype", "step")
@@ -1024,9 +1027,9 @@ def plot_data_model_comparison(
     xlabel: str | None = None,
     ylabel: str | None = None,
     data_label: str = "Data",
-    stacked_kwargs: dict | None = None,
-    unstacked_kwargs_list: list[dict] | None = None,
-    model_sum_kwargs: dict | None = None,
+    stacked_kwargs: dict[str, Any] | None = None,
+    unstacked_kwargs_list: list[dict[str, Any]] | None = None,
+    model_sum_kwargs: dict[str, Any] | None = None,
     model_uncertainty: bool = True,
     model_uncertainty_label: str = "Model stat. unc.",
     data_uncertainty_type: str = "asymmetrical",
@@ -1061,11 +1064,11 @@ def plot_data_model_comparison(
         The label for the y-axis. Default is None.
     data_label : str, optional
         The label for the data. Default is "Data".
-    stacked_kwargs : dict | None, optional
+    stacked_kwargs : dict[str, Any] | None, optional
         The keyword arguments used when plotting the stacked components in plot_hist() or plot_function(), one of which is called only once. Default is None.
-    unstacked_kwargs_list : list[dict] | None, optional
+    unstacked_kwargs_list : list[dict[str, Any]] | None, optional
         The list of keyword arguments used when plotting the unstacked components in plot_hist() or plot_function(), one of which is called once for each unstacked component. Default is None.
-    model_sum_kwargs : dict | None, optional
+    model_sum_kwargs : dict[str, Any] | None, optional
         The keyword arguments for the plot_hist() function for the sum of the model components.
         Has no effect if all the model components are stacked or if the model is one unstacked element.
         The special keyword "show" can be used with a boolean to specify whether to show or not the sum of the model components.
@@ -1138,11 +1141,11 @@ def plot_data_model_comparison(
         if plot_only is None:
             fig, (ax_main, ax_comparison) = create_comparison_figure()
         elif plot_only == "ax_main":
-            fig, ax_main = plt.subplots()
             _, ax_comparison = plt.subplots()
+            fig, ax_main = plt.subplots()
         elif plot_only == "ax_comparison":
-            fig, ax_comparison = plt.subplots()
             _, ax_main = plt.subplots()
+            fig, ax_comparison = plt.subplots()
         else:
             raise ValueError("plot_only must be 'ax_main', 'ax_comparison' or None.")
     elif fig is None or ax_main is None or ax_comparison is None:

plothist/variable_registry.py

@@ -6,8 +6,10 @@ from __future__ import annotations
 
 import os
 import warnings
+from typing import Any
 
 import boost_histogram as bh
+import numpy as np
 import yaml
 
 from plothist.histogramming import create_axis
@@ -36,14 +38,14 @@ def _check_if_variable_registry_exists(path: str) -> None:
 
 
 def _save_variable_registry(
-    variable_registry: dict[str, dict], path: str = "./variable_registry.yaml"
+    variable_registry: dict[str, dict[str, Any]], path: str = "./variable_registry.yaml"
 ) -> None:
     """
     Save the variable registry to a yaml file.
 
     Parameters
     ----------
-    variable_registry : dict[str, dict]
+    variable_registry : dict[str, dict[str, Any]]
         The variable registry to save.
     path : str, optional
         The path to the variable registry file (default is "./variable_registry.yaml").
@@ -62,7 +64,7 @@ def _save_variable_registry(
 def create_variable_registry(
     variable_keys: list[str],
     path: str = "./variable_registry.yaml",
-    custom_dict: dict | None = None,
+    custom_dict: dict[str, Any] | None = None,
     reset: bool = False,
 ) -> None:
     """
@@ -102,7 +104,7 @@ def create_variable_registry(
         A list of variable keys to be registered.
     path : str, optional
         The path to the variable registry file (default is "./variable_registry.yaml").
-    custom_dict : dict, optional
+    custom_dict : dict[str, Any], optional
         A dictionary containing the plotting information for the variables. Default dictionary is the one described above.
     reset : bool, optional
         If True, the registry will be reset to default values for all variable keys (default is False).
@@ -126,7 +128,7 @@ def create_variable_registry(
                         {
                             variable_key: {
                                 "name": variable_key,
-                                "bins": 50,
+                                "bins": "auto",
                                 "range": ("min", "max"),
                                 "label": variable_key,
                                 "log": False,
@@ -142,7 +144,7 @@ def create_variable_registry(
 
 def get_variable_from_registry(
     variable_key: str, path: str = "./variable_registry.yaml"
-) -> dict:
+) -> dict[str, Any]:
     """
     This function retrieves the parameter information for a variable from the variable registry file specified by the 'path' parameter.
     It loads the variable registry file and returns the dictionary entry corresponding to the specified variable name.
@@ -156,7 +158,7 @@ def get_variable_from_registry(
 
     Returns
     -------
-    dict
+    dict[str, Any]
         A dictionary containing the parameter information for the specified variable.
 
     See also
@@ -168,17 +170,26 @@ def get_variable_from_registry(
 
     with open(path) as f:
         variable_registry = yaml.safe_load(f)
-        if "range" in variable_registry[variable_key] and isinstance(
-            variable_registry[variable_key]["range"], list
-        ):
-            variable_registry[variable_key]["range"] = tuple(
-                variable_registry[variable_key]["range"]
-            )
-        return variable_registry[variable_key]
+
+    if not isinstance(variable_registry, dict):
+        raise RuntimeError(
+            f"Invalid registry format in {path}. Got {type(variable_registry)} instead of dict."
+        )
+
+    variable = variable_registry.get(variable_key)
+    if not isinstance(variable, dict):
+        raise RuntimeError(
+            f"Variable {variable_key} in {path} is not found or is not properly formatted."
+        )
+
+    if "range" in variable and isinstance(variable["range"], list):
+        variable["range"] = tuple(variable["range"])
+
+    return variable
 
 
 def update_variable_registry(
-    dictionary: dict,
+    dictionary: dict[str, Any],
     variable_keys: list[str] | None = None,
     path: str = "./variable_registry.yaml",
     overwrite: bool = False,
@@ -188,7 +199,7 @@ def update_variable_registry(
 
     Parameters
     ----------
-    dictionary : dict
+    dictionary : dict[str, Any]
         A dictionary containing the information to update the registry with.
     variable_keys : list[str]
         A list of variable keys for which to update the registry. Default is None: all variables in the registry are updated.
@@ -259,25 +270,37 @@ def remove_variable_registry_parameters(
     _save_variable_registry(variable_registry, path=path)
 
 
-def update_variable_registry_ranges(
+def update_variable_registry_ranges(*args, **kwargs):
+    warnings.warn(
+        "`update_variable_registry_ranges` is deprecated since v1.7.0 and will be removed in future versions. "
+        "Use `update_variable_registry_binning` instead.",
+        DeprecationWarning,
+        stacklevel=2,
+    )
+    return update_variable_registry_binning(*args, **kwargs)
+
+
+def update_variable_registry_binning(
     data,
     variable_keys: list[str] | None = None,
     path: str = "./variable_registry.yaml",
     overwrite: bool = False,
 ) -> None:
     """
-    Update the range parameters for multiple variables in the variable registry file.
+    Update both the bins and range parameters for multiple variables in the variable registry file.
 
     Parameters
     ----------
     data : numpy.ndarray or pandas.DataFrame
         A dataset containing the data for the variables.
     variable_keys : list[str]
-        A list of variable keys for which to update the range parameters in the registry. The variable needs to have a bin and range properties in the registry. Default is None: all variables in the registry are updated.
+        A list of variable keys for which to update the parameters in the registry.
+        The variable needs to have a bin and range properties in the registry.
+        Default is None: all variables in the registry are updated.
     path : str, optional
         The path to the variable registry file (default is "./variable_registry.yaml").
     overwrite : bool, optional
-        If True, the range parameters will be overwritten even if it's not equal to ("min", "max") (default is False).
+        If True, the bin and range parameters will be overwritten even if they differ from "auto" and ("min", "max") (default is False).
 
     Returns
     -------
@@ -302,13 +325,23 @@ def update_variable_registry_ranges(
                 f"Variable {variable_key} does not have a name, bins or range property in the registry {path}."
             )
 
-        range = ("min", "max") if overwrite else variable["range"]
+        bins = "auto" if overwrite else variable["bins"]
+        bin_number = len(np.histogram_bin_edges(data[variable["name"]], bins=bins)) - 1
 
-        if tuple(range) == ("min", "max"):
-            axis = create_axis(variable["bins"], tuple(range), data[variable["name"]])
+        range_val = ("min", "max") if overwrite else variable["range"]
+
+        if bins == "auto" or tuple(range_val) == ("min", "max"):
+            axis = create_axis(
+                bin_number,
+                tuple(range_val),
+                data[variable["name"]],
+            )
             if isinstance(axis, bh.axis.Regular):
                 update_variable_registry(
-                    {"range": (float(axis.edges[0]), float(axis.edges[-1]))},
+                    {
+                        "bins": bin_number,
+                        "range": (float(axis.edges[0]), float(axis.edges[-1])),
+                    },
                     [variable_key],
                     path=path,
                     overwrite=True,

{plothist-1.6.0.dist-info → plothist-1.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plothist
-Version: 1.6.0
+Version: 1.8.0
 Summary: Plot histograms in a scalable way and a beautiful style.
 Project-URL: Homepage, https://github.com/cyrraz/plothist
 Project-URL: Documentation, https://plothist.readthedocs.io/
@@ -14,14 +14,19 @@ Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: BSD License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Scientific/Engineering
-Requires-Python: >=3.9
+Requires-Python: >=3.10
 Requires-Dist: boost-histogram>=1.4.0
-Requires-Dist: matplotlib>=3.0
+Requires-Dist: matplotlib>=3.10.8
 Requires-Dist: numpy>=1.14.5
 Requires-Dist: plothist-utils>=0.0.1
-Requires-Dist: pyyaml>=5.3.1
-Requires-Dist: requests>=2.25.0
+Requires-Dist: pyyaml>=6.0.3
 Requires-Dist: scipy>=1.6.0
 Description-Content-Type: text/x-rst
 
@@ -46,9 +51,9 @@ plothist
    :width: 320
 
 
-|GitHub Project| |PyPI version| |Docs from main| |Discussion| |DOI| |Linter|
+|GitHub Project| |PyPI version| |conda version| |Docs from main| |Discussion| |DOI|
 
-|GitHub Actions Status: CI| |GitHub Actions Status: CD| |pre-commit.ci Status| |Code Coverage|
+|GitHub Actions Status: CI| |GitHub Actions Status: CD| |pre-commit.ci Status| |Linter| |Code Coverage|
 
 This package is a wrapper around `matplotlib <https://matplotlib.org/>`_.
 
@@ -75,6 +80,8 @@ This package is a wrapper around `matplotlib <https://matplotlib.org/>`_.
    :target: https://github.com/cyrraz/plothist
 .. |PyPI version| image:: https://badge.fury.io/py/plothist.svg?style=flat-square
    :target: https://badge.fury.io/py/plothist
+.. |conda version| image:: https://img.shields.io/conda/vn/conda-forge/plothist?style=platic
+   :target: https://anaconda.org/conda-forge/plothist
 .. |Docs from main| image:: https://img.shields.io/badge/docs-main-blue.svg?style=platic
    :target: https://plothist.readthedocs.io/en/main/
 .. |Discussion| image:: https://img.shields.io/static/v1?label=Discussions&message=Ask&color=blue&logo=github?style=flat-square

{plothist-1.6.0.dist-info → plothist-1.8.0.dist-info}/RECORD

@@ -1,13 +1,13 @@
-plothist/__init__.py,sha256=ikcYOZqMs4WKAJ2rM-3BnHu_CQxLlQCuc-iFo_a-Mns,2760
-plothist/_version.py,sha256=5FGJNp9Lkk9uOxeCjXpoCGBF79Ar6LGPOR7-atBqb_4,511
+plothist/__init__.py,sha256=-cUpXXg2cCdfjcbYAQWHLUHq-5kLsxSWgEbgtZYjqBA,2817
+plothist/_version.py,sha256=phQCmpksYlbZjXJhFI2j2KILuWjD0mXWO3Hg8g_EC8c,704
 plothist/_version.pyi,sha256=o7uNL6MhuJoiqpEnriU7rBT6TmkJZA-i2qMoNz9YcgQ,82
-plothist/comparison.py,sha256=jkONOiCp8RNBk8V_VvwACdfJJqo89rK9NQW71XxJscc,19623
+plothist/comparison.py,sha256=AUYvJBnG9ifUtxC-1N5JwzR7AX3MFTX2bCfuVCqkciM,19629
 plothist/default_style.mplstyle,sha256=7MmB2uiXmD_DSqFHeH1xxC-lTctBD_EASxMdSOsPep0,1574
-plothist/histogramming.py,sha256=o0RLdoEhxHA7Ws0bvK3DewsbQEu-QOtJNw5Md5AckHM,11474
-plothist/plothist_style.py,sha256=bbYFc-qsrTfdbacCBAAzmiHJoRIh3toBomFfxtTRNMs,13255
-plothist/plotters.py,sha256=ncPZa34Fo1mbFxv0guFc2zEciX_JXGKJxeIfcUDaP6w,46336
+plothist/histogramming.py,sha256=DJ9pl4i5cbvvkIsWs6MO6mMi_rAh_G94A1Nz8-v8zEs,11810
+plothist/plothist_style.py,sha256=u3hA1SAE9UebEnzluEn_xcs81hjYLlNDXS1OvkIm110,13782
+plothist/plotters.py,sha256=olfZOVMiuYGLnGOGnj0GxlI-Mufafwl_WvUNL-vZHxs,46782
 plothist/test_helpers.py,sha256=JXhxUdqMszkawxkU8cDPqipkSXNHfsKSfe3K-4frDrM,1379
-plothist/variable_registry.py,sha256=usVLoDLbIMzjeRDazIH0OIjQIUsh9RSJ9Ncnhn1V9qw,10783
+plothist/variable_registry.py,sha256=eOK_Df8vkwlj91J2YbRF3_7jB1NEGwlux8kOZULStkw,11867
 plothist/examples/README.rst,sha256=PVzkOQzVnNeWORxEv3NNv5XoSN2Y71D7SYzRkCqirfA,151
 plothist/examples/1d_hist/1d_comparison_asymmetry.py,sha256=Rp9gv6E5z9KgIhnraS9AjGSMgHLhSHd_u5L-vG0w-gU,757
 plothist/examples/1d_hist/1d_comparison_difference.py,sha256=FMKoTjMIQp6VsFXspVBy2AdxHPDnxhMfj9PCvE4qVjk,877
@@ -26,16 +26,16 @@ plothist/examples/1d_hist/1d_profile.py,sha256=otmerm64DNAYnRWaablQkQDgUVOvjLNDc
 plothist/examples/1d_hist/1d_side_by_side.py,sha256=sAHhBVnopINQscFaUpVI4KNs18e_1laHyrYBOlmU_fQ,1557
 plothist/examples/1d_hist/1d_str_category.py,sha256=2DtxfNMAlcZ6pYx44lP0-sgY4M73-R0aBH61uEJmKA0,880
 plothist/examples/1d_hist/README.rst,sha256=GT4-w9W-TDtjOqibOPcd6vYzAGO8bDsgxBRp3fD1uTw,100
-plothist/examples/2d_hist/2d_hist_correlations.py,sha256=OroqVDKjNlAiMVXZ1AvFarV3qESiFkL0dnnb_NFkI70,1834
+plothist/examples/2d_hist/2d_hist_correlations.py,sha256=c9PMymxNn9Kz0STvDQ3Pro4ndv-1PVG5RZsJCS9gobA,1836
 plothist/examples/2d_hist/2d_hist_simple.py,sha256=1n5yqOJxHzD31njOk85dTjuXhkogCDIS4-EG63XcnQ0,521
 plothist/examples/2d_hist/2d_hist_simple_discrete_colormap.py,sha256=E1zeMIDP25R_ppkB-wDz7TxaLsxf7dCs87FrPTPWTAc,1000
 plothist/examples/2d_hist/2d_hist_uneven.py,sha256=kR53XrSx38efG1wHEI69BIcPKVQzJygkr7ax_CFzBIA,646
 plothist/examples/2d_hist/2d_hist_with_projections.py,sha256=EPGNoXQu7Oa0xS_fDc21_0dCVPLeKpSyv3dylfwXIDs,725
 plothist/examples/2d_hist/README.rst,sha256=SCD39UgHb8vat63gNkOlDWPjwZj9INUS516D5DxW0Vk,116
 plothist/examples/advanced/1d_comparison_advanced.py,sha256=RHWejmv8OQcgZ2R9yfAehv8cc8vQ5qr5gcVApGWdQhI,2668
-plothist/examples/advanced/1d_side_by_side_with_numbers.py,sha256=46k1RXFPQ9x2yfeWQdAh5sz4I9C0SnYtQgqeL29_oQY,2437
+plothist/examples/advanced/1d_side_by_side_with_numbers.py,sha256=-6KGJ48S6CT1ZM-krROUtAWZDrLqDhuQ6Ue5lRMeWBo,2463
 plothist/examples/advanced/README.rst,sha256=ulygCIs7LUaViLpeTCIG2qcG9FEohccrlR2nXYLnkO4,112
-plothist/examples/advanced/asymmetry_comparison_advanced.py,sha256=fteFmenSzhhZxgz4OU3fL8ejZKGnGpbNz-Zmt4M6g2c,2670
+plothist/examples/advanced/asymmetry_comparison_advanced.py,sha256=9VvYvCkc7smjyuEQazxqo25xl0P5wzAugbW5ATjXDto,2741
 plothist/examples/advanced/model_examples_flatten2D.py,sha256=b4Rl6ItrsaddVL5uC9Gr3ccgSyhjW1RAtxSQgcCs_JA,2233
 plothist/examples/func_1d/README.rst,sha256=tez3Jpva9yv7T7bDfXE9dCdmThJ1ALZRiEHOfFkrCNk,108
 plothist/examples/func_1d/fct_1d.py,sha256=DRbKnImIqVYVXqH79X54YnxyGLu914fEmxry3fLBDsM,455
@@ -54,11 +54,11 @@ plothist/examples/model_ex/ratio_data_vs_model_with_stacked_and_unstacked_functi
 plothist/examples/utility/README.rst,sha256=8qQeSXjzdPDlp58cbaHnAq5O55BSp8V-Umr9wmouwJ8,77
 plothist/examples/utility/add_text_example.py,sha256=EB7ZUStt2E8GawjWNmVbSV34NINZ8v-4-rQBw1_gFD4,811
 plothist/examples/utility/color_palette_hists.py,sha256=uIc6TrmjTj8EUif1Yj1wq4nXoY1sgJpS15yPe3-FpTw,2383
-plothist/examples/utility/color_palette_squares.py,sha256=pNasgvZZApu-sCqhi5FTJAH3l6Px2cgB5RwJcQ1eF1U,2689
+plothist/examples/utility/color_palette_squares.py,sha256=kBUhoEa0TM5dzDEhUiPkVrOmcZFmts4NqA5ow_4yyZ0,2779
 plothist/examples/utility/matplotlib_vs_plothist_style.py,sha256=iEIfbFKmAN-PcDCzw_sBuhrfpes4I2wUVfC0r1vUHEw,2006
 plothist/examples/utility/uncertainty_types.py,sha256=r1yxzEvxn5Clv0ZIok9JjVh3RB-YQ_usWhFnpvl1els,3259
-plothist-1.6.0.dist-info/METADATA,sha256=WsoMGQrcmUasAx0UxEP9aGEw7LAk1eHQCrjUz6228fY,4737
-plothist-1.6.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-plothist-1.6.0.dist-info/licenses/AUTHORS.md,sha256=02x3_8PNyTsXcRs0IlJeCTOmpGNRqymcJ71-2QtR37E,111
-plothist-1.6.0.dist-info/licenses/LICENSE,sha256=bfaEdGehofQDaw-zDdVMHNUKo1FrOm6oGUEF-ltrp6w,1523
-plothist-1.6.0.dist-info/RECORD,,
+plothist-1.8.0.dist-info/METADATA,sha256=jhJb4Vi2o11WwUbWdVoFjaDHLmYwDUuq8I525gQKwH8,5175
+plothist-1.8.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+plothist-1.8.0.dist-info/licenses/AUTHORS.md,sha256=ngUOEAYO6Ed-wWStDZDemPZGRian7rVpTIqGUt_ECkE,115
+plothist-1.8.0.dist-info/licenses/LICENSE,sha256=uiArTxl1M6mNZcA2HV7Dhd06--H5C3kz0_qIa2oQXkU,1523
+plothist-1.8.0.dist-info/RECORD,,

{plothist-1.6.0.dist-info → plothist-1.8.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any

plothist-1.8.0.dist-info/licenses/AUTHORS.md

@@ -0,0 +1,2 @@
+- Cyrille Praz ([@cyrraz](https://github.com/cyrraz))
+- Tristan Fillinger ([@0ctagon](https://github.com/0ctagon))

{plothist-1.6.0.dist-info → plothist-1.8.0.dist-info}/licenses/LICENSE

@@ -1,6 +1,6 @@
 BSD 3-Clause License
 
-Copyright (c) 2023-2025, Cyrille Praz, Tristan Fillinger
+Copyright (c) 2023-2026, Cyrille Praz, Tristan Fillinger
 
 Redistribution and use in source and binary forms, with or without
 modification, are permitted provided that the following conditions are met:

plothist-1.6.0.dist-info/licenses/AUTHORS.md

@@ -1,2 +0,0 @@
-Cyrille Praz ([@cyrraz](https://github.com/cyrraz))
-Tristan Fillinger ([@0ctagon](https://github.com/0ctagon))