scitex 2.5.0__py3-none-any.whl → 2.7.3__py3-none-any.whl

scitex/plt/ax/_plot/_plot_statistical_shaded_line.py

@@ -65,16 +65,18 @@ def stx_line(axis, values_1d, xx=None, **kwargs):
     df : pandas.DataFrame
         DataFrame with x and y values
     """
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
     values_1d = np.asarray(values_1d)
     assert values_1d.ndim <= 2, f"Data must be 1D or 2D, got {values_1d.ndim}D"
     if xx is None:
         xx = np.arange(len(values_1d))
     else:
         xx = np.asarray(xx)
-    assert len(xx) == len(
-        values_1d
-    ), f"xx length ({len(xx)}) must match values_1d length ({len(values_1d)})"
+    assert len(xx) == len(values_1d), (
+        f"xx length ({len(xx)}) must match values_1d length ({len(values_1d)})"
+    )
 
     axis.plot(xx, values_1d, **kwargs)
     return axis, pd.DataFrame({"x": xx, "y": values_1d})
@@ -103,7 +105,9 @@ def stx_mean_std(axis, values_2d, xx=None, sd=1, **kwargs):
     axis : matplotlib.axes.Axes or scitex.plt._subplots.AxisWrapper
         The axis with the plot
     """
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
     assert isinstance(sd, (int, float)), f"sd must be a number, got {type(sd)}"
     assert sd >= 0, f"sd must be non-negative, got {sd}"
     values_2d = np.asarray(values_2d)
@@ -113,9 +117,9 @@ def stx_mean_std(axis, values_2d, xx=None, sd=1, **kwargs):
     else:
         xx = np.asarray(xx)
     expected_len = values_2d.shape[1] if values_2d.ndim > 1 else len(values_2d)
-    assert (
-        len(xx) == expected_len
-    ), f"xx length ({len(xx)}) must match values_2d length ({expected_len})"
+    assert len(xx) == expected_len, (
+        f"xx length ({len(xx)}) must match values_2d length ({expected_len})"
+    )
 
     if values_2d.ndim == 1:
         central = values_2d
@@ -157,10 +161,10 @@ def stx_mean_ci(axis, values_2d, xx=None, perc=95, **kwargs):
     axis : matplotlib.axes.Axes or scitex.plt._subplots.AxisWrapper
         The axis with the plot
     """
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
-    assert isinstance(
-        perc, (int, float)
-    ), f"perc must be a number, got {type(perc)}"
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
+    assert isinstance(perc, (int, float)), f"perc must be a number, got {type(perc)}"
     assert 0 <= perc <= 100, f"perc must be between 0 and 100, got {perc}"
     values_2d = np.asarray(values_2d)
     assert values_2d.ndim <= 2, f"Data must be 1D or 2D, got {values_2d.ndim}D"
@@ -171,9 +175,9 @@ def stx_mean_ci(axis, values_2d, xx=None, perc=95, **kwargs):
         xx = np.asarray(xx)
 
     expected_len = values_2d.shape[1] if values_2d.ndim > 1 else len(values_2d)
-    assert (
-        len(xx) == expected_len
-    ), f"xx length ({len(xx)}) must match values_2d length ({expected_len})"
+    assert len(xx) == expected_len, (
+        f"xx length ({len(xx)}) must match values_2d length ({expected_len})"
+    )
 
     if values_2d.ndim == 1:
         central = values_2d
@@ -216,7 +220,9 @@ def stx_median_iqr(axis, values_2d, xx=None, **kwargs):
     axis : matplotlib.axes.Axes or scitex.plt._subplots.AxisWrapper
         The axis with the plot
     """
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
     values_2d = np.asarray(values_2d)
     assert values_2d.ndim <= 2, f"Data must be 1D or 2D, got {values_2d.ndim}D"
 
@@ -226,9 +232,9 @@ def stx_median_iqr(axis, values_2d, xx=None, **kwargs):
         xx = np.asarray(xx)
 
     expected_len = values_2d.shape[1] if values_2d.ndim > 1 else len(values_2d)
-    assert (
-        len(xx) == expected_len
-    ), f"xx length ({len(xx)}) must match values_2d length ({expected_len})"
+    assert len(xx) == expected_len, (
+        f"xx length ({len(xx)}) must match values_2d length ({expected_len})"
+    )
 
     if values_2d.ndim == 1:
         central = values_2d

scitex/plt/ax/_plot/_stx_conf_mat.py

@@ -41,7 +41,7 @@ def stx_conf_mat(
 
     Parameters
     ----------
-    axis : plt.Axes or scitex.plt._subplots._AxisWrapper.AxisWrapper
+    axis : plt.Axes or scitex.plt._subplots.AxisWrapper
         Matplotlib axes or scitex axis wrapper to plot on
     conf_mat_2d : Union[np.ndarray, pd.DataFrame], shape (n_classes, n_classes)
         2D confusion matrix data (true labels × predicted labels)
@@ -68,7 +68,7 @@ def stx_conf_mat(
 
     Returns
     -------
-    Union[plt.Axes, Tuple[plt.Axes, float]] or Union[scitex.plt._subplots._AxisWrapper.AxisWrapper, Tuple[scitex.plt._subplots._AxisWrapper.AxisWrapper, float]]
+    Union[plt.Axes, Tuple[plt.Axes, float]] or Union[scitex.plt._subplots.AxisWrapper, Tuple[scitex.plt._subplots.AxisWrapper, float]]
         Axes object and optionally balanced accuracy
 
     Example
@@ -81,7 +81,9 @@ def stx_conf_mat(
     Balanced Accuracy: 0.889
     """
 
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
 
     if not isinstance(conf_mat_2d, pd.DataFrame):
         conf_mat_2d = pd.DataFrame(conf_mat_2d)
@@ -133,4 +135,5 @@ def stx_conf_mat(
     else:
         return axis, None
 
+
 # EOF

scitex/plt/ax/_plot/_stx_ecdf.py

@@ -5,16 +5,18 @@
 
 """Empirical Cumulative Distribution Function (ECDF) plotting."""
 
-import warnings
 from typing import Any, Tuple, Union
 
 import numpy as np
 import pandas as pd
 from matplotlib.axes import Axes
 
+from scitex import logging
 from scitex.pd._force_df import force_df as scitex_pd_force_df
 from ....plt.utils import assert_valid_axis, mm_to_pt
 
+logger = logging.getLogger(__name__)
+
 
 # Default line width (0.2mm for publication)
 DEFAULT_LINE_WIDTH_MM = 0.2
@@ -59,14 +61,16 @@ def stx_ecdf(
     >>> fig, ax = stx.plt.subplots()
     >>> ax, df = stx.plt.ax.stx_ecdf(ax, data)
     """
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
 
     # Flatten and remove NaN values
     values_1d = np.hstack(values_1d)
 
     # Warnings
     if np.isnan(values_1d).any():
-        warnings.warn("NaN value are ignored for ECDF plot.")
+        logger.warning("NaN values are ignored for ECDF plot.")
     values_1d = values_1d[~np.isnan(values_1d)]
     nn = len(values_1d)
 
@@ -79,8 +83,8 @@ def stx_ecdf(
     y_step = np.repeat(ecdf_perc, 2)[:-1]
 
     # Apply default linewidth if not specified
-    if 'linewidth' not in kwargs and 'lw' not in kwargs:
-        kwargs['linewidth'] = mm_to_pt(DEFAULT_LINE_WIDTH_MM)
+    if "linewidth" not in kwargs and "lw" not in kwargs:
+        kwargs["linewidth"] = mm_to_pt(DEFAULT_LINE_WIDTH_MM)
 
     # Add sample size to label if provided
     if "label" in kwargs and kwargs["label"]:

scitex/plt/ax/_plot/_stx_fillv.py

@@ -42,9 +42,11 @@ def stx_fillv(axes, starts_1d, ends_1d, color="red", alpha=0.2):
     axes = axes if isinstance(axes, np.ndarray) else [axes]
 
     for ax in axes:
-        assert_valid_axis(ax, "First argument must be a matplotlib axis or scitex axis wrapper")
+        assert_valid_axis(
+            ax, "First argument must be a matplotlib axis or scitex axis wrapper"
+        )
         for start, end in zip(starts_1d, ends_1d):
-            ax.axvspan(start, end, facecolor=color, edgecolor='none', alpha=alpha)
+            ax.axvspan(start, end, facecolor=color, edgecolor="none", alpha=alpha)
 
     if not is_axes:
         return axes[0]

scitex/plt/ax/_plot/_stx_heatmap.py

@@ -198,6 +198,7 @@ def _mpl_heatmap(
 
     # Format colorbar ticks
     from matplotlib.ticker import MaxNLocator
+
     cbar.ax.yaxis.set_major_locator(MaxNLocator(nbins=4, min_n_ticks=3))
     cbar.ax.tick_params(width=0.2 * 2.83465, length=0.8 * 2.83465)  # Match tick styling
 
@@ -218,7 +219,7 @@ def _mpl_heatmap(
     ax.spines[:].set_visible(True)
 
     # Set aspect ratio to 'equal' for square cells (1:1)
-    ax.set_aspect('equal', adjustable='box')
+    ax.set_aspect("equal", adjustable="box")
 
     ax.set_xticks(np.arange(data.shape[1] + 1) - 0.5, minor=True)
     ax.set_yticks(np.arange(data.shape[0] + 1) - 0.5, minor=True)
@@ -317,7 +318,9 @@ def _mpl_annotate_heatmap(
 
     # Set default alignment to center, but allow it to be
     # overwritten by textkw.
-    kw = dict(horizontalalignment="center", verticalalignment="center", fontsize=fontsize)
+    kw = dict(
+        horizontalalignment="center", verticalalignment="center", fontsize=fontsize
+    )
     kw.update(textkw)
 
     # Get the formatter in case a string is supplied
@@ -343,8 +346,8 @@ if __name__ == "__main__":
     import numpy as np
 
     data = np.random.rand(5, 10)
-    x_labels = [f"X{ii+1}" for ii in range(5)]
-    y_labels = [f"Y{ii+1}" for ii in range(10)]
+    x_labels = [f"X{ii + 1}" for ii in range(5)]
+    y_labels = [f"Y{ii + 1}" for ii in range(10)]
 
     fig, ax = plt.subplots()
 

scitex/plt/ax/_plot/_stx_image.py

@@ -31,10 +31,10 @@ def stx_image(
     Imshows an two-dimensional array with theese two conditions:
     1) The first dimension represents the x dim, from left to right.
     2) The second dimension represents the y dim, from bottom to top
-    
+
     Parameters
     ----------
-    ax : matplotlib.axes.Axes or scitex.plt._subplots._AxisWrapper.AxisWrapper
+    ax : matplotlib.axes.Axes or scitex.plt._subplots.AxisWrapper
         The axis to plot on
     arr_2d : numpy.ndarray
         The 2D array to display
@@ -58,13 +58,15 @@ def stx_image(
         Maximum data value for colormap scaling, by default None
     **kwargs
         Additional keyword arguments passed to ax.imshow()
-        
+
     Returns
     -------
-    matplotlib.axes.Axes or scitex.plt._subplots._AxisWrapper.AxisWrapper
+    matplotlib.axes.Axes or scitex.plt._subplots.AxisWrapper
         The axis with the image plotted
     """
-    assert_valid_axis(ax, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        ax, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
     assert arr_2d.ndim == 2, "Input array must be 2-dimensional"
 
     if kwargs.get("xyz"):

scitex/plt/ax/_plot/_stx_joyplot.py

@@ -15,8 +15,9 @@ from scipy import stats
 from ....plt.utils import assert_valid_axis
 
 
-def stx_joyplot(ax, arrays, overlap=0.5, fill_alpha=0.7, line_alpha=1.0,
-                 colors=None, **kwargs):
+def stx_joyplot(
+    ax, arrays, overlap=0.5, fill_alpha=0.7, line_alpha=1.0, colors=None, **kwargs
+):
     """
     Create a joyplot (ridgeline plot) on the provided axes.
 
@@ -42,7 +43,13 @@ def stx_joyplot(ax, arrays, overlap=0.5, fill_alpha=0.7, line_alpha=1.0,
     matplotlib.axes.Axes
         The axes with the joyplot
     """
-    assert_valid_axis(ax, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        ax, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
+
+    # Convert dict to list of arrays (values only)
+    if isinstance(arrays, dict):
+        arrays = list(arrays.values())
 
     # Add sample size per distribution to label if provided (show range if variable)
     if kwargs.get("label"):
@@ -57,8 +64,14 @@ def stx_joyplot(ax, arrays, overlap=0.5, fill_alpha=0.7, line_alpha=1.0,
     # Default colors from scitex palette
     if colors is None:
         colors = [
-            HEX["blue"], HEX["red"], HEX["green"], HEX["yellow"],
-            HEX["purple"], HEX["orange"], HEX["lightblue"], HEX["pink"],
+            HEX["blue"],
+            HEX["red"],
+            HEX["green"],
+            HEX["yellow"],
+            HEX["purple"],
+            HEX["orange"],
+            HEX["lightblue"],
+            HEX["pink"],
         ]
 
     n_ridges = len(arrays)
@@ -92,14 +105,23 @@ def stx_joyplot(ax, arrays, overlap=0.5, fill_alpha=0.7, line_alpha=1.0,
         baseline = i * (1.0 - overlap)
 
         # Scale density to fit nicely
-        scaled_density = kdes[i] / max_density * ridge_height if max_density > 0 else kdes[i]
+        scaled_density = (
+            kdes[i] / max_density * ridge_height if max_density > 0 else kdes[i]
+        )
 
         # Fill
-        ax.fill_between(x, baseline, baseline + scaled_density,
-                       facecolor=color, edgecolor='none', alpha=fill_alpha)
+        ax.fill_between(
+            x,
+            baseline,
+            baseline + scaled_density,
+            facecolor=color,
+            edgecolor="none",
+            alpha=fill_alpha,
+        )
         # Line on top
-        ax.plot(x, baseline + scaled_density, color=color, alpha=line_alpha,
-               linewidth=1.0)
+        ax.plot(
+            x, baseline + scaled_density, color=color, alpha=line_alpha, linewidth=1.0
+        )
 
     # Set y limits
     ax.set_ylim(-0.1, n_ridges * (1.0 - overlap) + ridge_height)

scitex/plt/ax/_plot/_stx_raster.py

@@ -31,7 +31,7 @@ def stx_raster(
     apply_set_n_ticks=True,
     n_xticks=4,
     n_yticks=None,
-    **kwargs
+    **kwargs,
 ):
     """
     Create a raster plot using eventplot with custom labels and colors.
@@ -72,7 +72,9 @@ def stx_raster(
     df : pandas.DataFrame
         DataFrame with time indices and channel events.
     """
-    assert_valid_axis(ax, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        ax, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
 
     # Format spike_times_list data
     spike_times_list = _ensure_list(spike_times_list)
@@ -99,13 +101,24 @@ def stx_raster(
     if np.isscalar(lineoffsets):
         lineoffsets = [lineoffsets]
     if len(lineoffsets) < len(spike_times_list):
-        lineoffsets = list(lineoffsets) + list(range(len(lineoffsets), len(spike_times_list)))
+        lineoffsets = list(lineoffsets) + list(
+            range(len(lineoffsets), len(spike_times_list))
+        )
 
     # Plotting as eventplot using spike_times_list with proper positioning
-    for ii, (pos, color, offset) in enumerate(zip(spike_times_list, colors, lineoffsets)):
+    for ii, (pos, color, offset) in enumerate(
+        zip(spike_times_list, colors, lineoffsets)
+    ):
         label = _define_label(labels, ii)
-        ax.eventplot(pos, lineoffsets=offset, linelengths=linelengths,
-                    orientation=orientation, colors=color, label=label, **kwargs)
+        ax.eventplot(
+            pos,
+            lineoffsets=offset,
+            linelengths=linelengths,
+            orientation=orientation,
+            colors=color,
+            label=label,
+            **kwargs,
+        )
 
     # Apply set_n_ticks for cleaner axes if requested
     if apply_set_n_ticks:
@@ -114,20 +127,20 @@ def stx_raster(
         # For categorical y-axis (trials/channels), use appropriate tick count
         if n_yticks is None:
             n_yticks = min(len(spike_times_list), 8)  # Max 8 ticks for readability
-        
+
         # Only apply if we have reasonable numeric ranges
         try:
             x_range = ax.get_xlim()
             y_range = ax.get_ylim()
-            
+
             # Apply x-ticks if we have a reasonable numeric range
             if x_range[1] - x_range[0] > 0:
                 set_n_ticks(ax, n_xticks=n_xticks, n_yticks=None)
-            
+
             # Apply y-ticks only if we don't have categorical labels
             if labels is None and y_range[1] - y_range[0] > 0:
                 set_n_ticks(ax, n_xticks=None, n_yticks=n_yticks)
-                
+
         except Exception:
             # Skip set_n_ticks if there are issues (e.g., categorical data)
             pass
@@ -137,7 +150,9 @@ def stx_raster(
         ax.legend()
 
     # Return spike_times in a useful format
-    spike_times_digital_df = _event_times_to_digital_df(spike_times_list, time, lineoffsets)
+    spike_times_digital_df = _event_times_to_digital_df(
+        spike_times_list, time, lineoffsets
+    )
 
     return ax, spike_times_digital_df
 

scitex/plt/ax/_plot/_stx_rectangle.py

@@ -61,8 +61,8 @@ def stx_rectangle(ax, xx, yy, ww, hh, **kwargs):
     matplotlib.axes.Axes.add_patch : Method used to add the patch
     """
     # Default to no edge for cleaner publication figures
-    if 'edgecolor' not in kwargs and 'ec' not in kwargs:
-        kwargs['edgecolor'] = 'none'
+    if "edgecolor" not in kwargs and "ec" not in kwargs:
+        kwargs["edgecolor"] = "none"
     ax.add_patch(Rectangle((xx, yy), ww, hh, **kwargs))
     return ax
 

scitex/plt/ax/_plot/_stx_shaded_line.py

@@ -53,10 +53,12 @@ def _plot_single_shaded_line(
     df : pd.DataFrame
         DataFrame with x, y_lower, y_middle, y_upper columns.
     """
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
-    assert (
-        len(xx) == len(y_middle) == len(y_lower) == len(y_upper)
-    ), "All arrays must have the same length"
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
+    assert len(xx) == len(y_middle) == len(y_lower) == len(y_upper), (
+        "All arrays must have the same length"
+    )
 
     label = kwargs.pop("label", None)
     axis.plot(xx, y_middle, color=color, alpha=alpha, label=label, **kwargs)
@@ -104,10 +106,12 @@ def _plot_shaded_line(
     results : list of pd.DataFrame
         List of DataFrames with plot data.
     """
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
-    assert (
-        len(xs) == len(ys_lower) == len(ys_middle) == len(ys_upper)
-    ), "All input lists must have the same length"
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
+    assert len(xs) == len(ys_lower) == len(ys_middle) == len(ys_upper), (
+        "All input lists must have the same length"
+    )
 
     results = []
     colors = color
@@ -199,9 +203,9 @@ def stx_shaded_line(
     )
 
     if is_single:
-        assert (
-            len(xs) == len(ys_lower) == len(ys_middle) == len(ys_upper)
-        ), "All arrays must have the same length for single line plot"
+        assert len(xs) == len(ys_lower) == len(ys_middle) == len(ys_upper), (
+            "All arrays must have the same length for single line plot"
+        )
 
         return _plot_single_shaded_line(
             axis, xs, ys_lower, ys_middle, ys_upper, color=color, **kwargs

scitex/plt/ax/_plot/_stx_violin.py

@@ -105,7 +105,9 @@ def sns_plot_violin(ax, data=None, x=None, y=None, hue=None, half=False, **kwarg
     ax : matplotlib.axes.Axes or scitex.plt._subplots.AxisWrapper
         The axes object with the plot
     """
-    assert_valid_axis(ax, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        ax, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
 
     if not half:
         # Standard violin plot

scitex/plt/ax/_style/_add_marginal_ax.py

@@ -17,21 +17,23 @@ from ....plt.utils import assert_valid_axis
 def add_marginal_ax(axis, place, size=0.2, pad=0.1):
     """
     Add a marginal axis to the specified side of an existing axis.
-    
+
     Arguments:
         axis (matplotlib.axes.Axes or scitex.plt._subplots.AxisWrapper): The axis to which a marginal axis will be added.
         place (str): Where to place the marginal axis ('top', 'right', 'bottom', or 'left').
         size (float, optional): Fractional size of the marginal axis relative to the main axis. Defaults to 0.2.
         pad (float, optional): Padding between the axes. Defaults to 0.1.
-        
+
     Returns:
         matplotlib.axes.Axes: The newly created marginal axis.
     """
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
 
     divider = make_axes_locatable(axis)
 
-    size_perc_str = f"{size*100}%"
+    size_perc_str = f"{size * 100}%"
     if place in ["left", "right"]:
         size = 1.0 / size
 

scitex/plt/ax/_style/_auto_scale_axis.py

@@ -18,7 +18,9 @@ import numpy as np
 from typing import Optional, Tuple
 
 
-def detect_scale_factor(values: np.ndarray, threshold: float = 1e-2) -> Tuple[int, bool]:
+def detect_scale_factor(
+    values: np.ndarray, threshold: float = 1e-2
+) -> Tuple[int, bool]:
     """
     Detect appropriate power of 10 to factor out from axis values.
 
@@ -97,7 +99,7 @@ def format_scale_factor(power: int) -> str:
     return f"×10$^{{{power}}}$"
 
 
-def auto_scale_axis(ax, axis: str = 'both', threshold: float = 1e-2) -> None:
+def auto_scale_axis(ax, axis: str = "both", threshold: float = 1e-2) -> None:
     """
     Automatically scale axis to factor out common powers of 10.
 
@@ -155,16 +157,16 @@ def auto_scale_axis(ax, axis: str = 'both', threshold: float = 1e-2) -> None:
             return
 
         # Create scaling factor
-        scale_factor = 10 ** power
+        scale_factor = 10**power
 
         # Update tick formatter to show scaled values
         def format_func(value, pos):
             scaled_value = value / scale_factor
             # Format with appropriate precision
             if abs(scaled_value) < 10:
-                return f'{scaled_value:.1f}'
+                return f"{scaled_value:.1f}"
             else:
-                return f'{scaled_value:.0f}'
+                return f"{scaled_value:.0f}"
 
         set_formatter(ticker.FuncFormatter(format_func))
 
@@ -173,22 +175,24 @@ def auto_scale_axis(ax, axis: str = 'both', threshold: float = 1e-2) -> None:
         scale_str = format_scale_factor(power)
 
         # Check if label already has units in brackets
-        if '[' in current_label and ']' in current_label:
+        if "[" in current_label and "]" in current_label:
             # Insert scale factor before the closing bracket
             # e.g., "Density [a.u.]" → "Density [×10⁻³ a.u.]"
-            label_parts = current_label.rsplit(']', 1)
+            label_parts = current_label.rsplit("]", 1)
             new_label = f"{label_parts[0]} {scale_str}]{label_parts[1]}"
         else:
             # Append scale factor in brackets
             # e.g., "Density" → "Density [×10⁻³]"
-            new_label = f"{current_label} [{scale_str}]" if current_label else f"[{scale_str}]"
+            new_label = (
+                f"{current_label} [{scale_str}]" if current_label else f"[{scale_str}]"
+            )
 
         set_label(new_label)
 
     # Apply to requested axes
-    if axis in ['x', 'both']:
+    if axis in ["x", "both"]:
         scale_axis_impl(ax, is_x_axis=True)
-    if axis in ['y', 'both']:
+    if axis in ["y", "both"]:
         scale_axis_impl(ax, is_x_axis=False)
 
 

scitex/plt/ax/_style/_extend.py

@@ -29,7 +29,9 @@ def extend(axis, x_ratio=1.0, y_ratio=1.0):
         AssertionError: If the first argument is not a valid axis.
     """
 
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
 
     assert x_ratio != 0, "x_ratio must not be 0."
     assert y_ratio != 0, "y_ratio must not be 0."

scitex/plt/ax/_style/_force_aspect.py

@@ -16,15 +16,17 @@ from ....plt.utils import assert_valid_axis
 def force_aspect(axis, aspect=1):
     """
     Forces aspect ratio of an axis based on the extent of the image.
-    
+
     Arguments:
         axis (matplotlib.axes.Axes or scitex.plt._subplots.AxisWrapper): The axis to adjust.
         aspect (float, optional): The aspect ratio to apply. Defaults to 1.
-        
+
     Returns:
         matplotlib.axes.Axes or scitex.plt._subplots.AxisWrapper: The axis with adjusted aspect ratio.
     """
-    assert_valid_axis(axis, "First argument must be a matplotlib axis or scitex axis wrapper")
+    assert_valid_axis(
+        axis, "First argument must be a matplotlib axis or scitex axis wrapper"
+    )
 
     im = axis.get_images()
 

scitex/plt/ax/_style/_format_units.py

@@ -89,12 +89,12 @@ def format_label_auto(text: str) -> str:
 
     # Pattern to match units in parentheses at the end of the string
     # e.g., "Time (s)" or "Frequency (Hz)"
-    pattern = r'\s*\(([^)]+)\)\s*$'
+    pattern = r"\s*\(([^)]+)\)\s*$"
 
     match = re.search(pattern, text)
     if match:
         unit = match.group(1)
-        label = text[:match.start()].strip()
+        label = text[: match.start()].strip()
         return f"{label} [{unit}]"
 
     return text