tesorotools-python 0.0.26__py3-none-any.whl → 0.0.28__py3-none-any.whl

tesorotools/artists/stacked.py

@@ -139,6 +139,13 @@ class StackedBarPlot:
 
     Positive and negative values are stacked separately so
     that bars extend in both directions from the baseline.
+
+    The ``plot`` method delegates to overridable hooks so
+    subclasses can customise individual steps without
+    reimplementing the full render pipeline:
+
+    * ``_prepare_data`` — slice, scale, resample
+    * ``_format_xticks`` — tick positions, labels, rotation
     """
 
     def __init__(
@@ -156,6 +163,8 @@ class StackedBarPlot:
         figsize: tuple[float, float] | None = None,
         overlay_series: dict[str, str] | None = None,
         plot_size: tuple[float, float] | None = None,
+        bar_width: float = 0.7,
+        x_rotation: float = 0,
     ) -> None:
         if out_path.suffix != ".png":
             raise ValueError(f"out_path must be .png: {out_path}")
@@ -171,6 +180,8 @@ class StackedBarPlot:
         self.plot_size = plot_size
         self.figsize = figsize
         self.overlay_series = overlay_series or {}
+        self.bar_width = bar_width
+        self.x_rotation = x_rotation
 
     @classmethod
     def from_yaml(cls, loader: TemplateLoader, node: MappingNode) -> Self:
@@ -182,7 +193,13 @@ class StackedBarPlot:
         cfg["data"] = pd.read_feather(cfg.pop("data_path"))
         return cls(**cfg)
 
-    def plot(self) -> Axes:
+    # -- Overridable hooks -----------------------------------
+
+    def _prepare_data(self) -> pd.DataFrame:
+        """Slice, scale, and return plot-ready DataFrame.
+
+        Override to resample, change the index type, etc.
+        """
         start = (
             pd.Timestamp(self.start_date)
             if self.start_date
@@ -193,10 +210,35 @@ class StackedBarPlot:
             if self.end_date
             else self.data.index.max()
         )
-
         all_cols = list(self.series.keys()) + list(self.overlay_series.keys())
         plot_data = self.data.loc[start:end, all_cols].dropna()
-        plot_data = plot_data * self.scale
+        return plot_data * self.scale
+
+    def _format_xticks(
+        self,
+        ax: Axes,
+        plot_data: pd.DataFrame,
+        x: np.ndarray[tuple[int], np.dtype[np.intp]],
+    ) -> None:
+        """Set tick positions, labels, and rotation.
+
+        Override for custom labels (e.g. string index).
+        """
+        dates = plot_data.index
+        step = max(1, len(dates) // 12)
+        tick_pos = list(range(0, len(dates), step))
+        tick_labels = [dates[i].strftime("%Y") for i in tick_pos]
+        ax.set_xticks(  # type: ignore[reportUnknownMemberType]
+            tick_pos
+        )
+        ax.set_xticklabels(  # type: ignore[reportUnknownMemberType]
+            tick_labels, rotation=self.x_rotation
+        )
+
+    # -- Main render -----------------------------------------
+
+    def plot(self) -> Axes:
+        plot_data = self._prepare_data()
 
         fig_kw = dict(FIG_CONFIG)
         if self.figsize is not None:
@@ -210,7 +252,6 @@ class StackedBarPlot:
         labels = list(self.series.values())
 
         x = np.arange(len(plot_data))
-        bar_width = 0.7
 
         pos_bottom: np.ndarray[tuple[int], np.dtype[np.float64]] = np.zeros(
             len(plot_data)
@@ -235,7 +276,7 @@ class StackedBarPlot:
                     x,
                     pos,
                     bottom=pos_bottom,
-                    width=bar_width,
+                    width=self.bar_width,
                     label=label,
                 )
                 .patches[0]
@@ -245,7 +286,7 @@ class StackedBarPlot:
                 x,
                 neg,
                 bottom=neg_bottom,
-                width=bar_width,
+                width=self.bar_width,
                 color=color,
             )
             pos_bottom = pos_bottom + pos
@@ -262,16 +303,7 @@ class StackedBarPlot:
                 zorder=10,
             )
 
-        dates = plot_data.index
-        step = max(1, len(dates) // 12)
-        tick_pos = list(range(0, len(dates), step))
-        tick_labels = [dates[i].strftime("%Y") for i in tick_pos]
-        ax.set_xticks(  # type: ignore[reportUnknownMemberType]
-            tick_pos
-        )
-        ax.set_xticklabels(  # type: ignore[reportUnknownMemberType]
-            tick_labels
-        )
+        self._format_xticks(ax, plot_data, x)
 
         style_spines(
             ax,

tesorotools/pipeline/engine.py

@@ -51,6 +51,12 @@ def apply_transformations(
 ) -> pd.DataFrame:
     """Apply rules sequentially to a DataFrame.
 
+    New columns are accumulated in a dict and merged into
+    *result* only when a later rule reads from them (or at
+    the end).  This avoids per-rule fragmentation of the
+    DataFrame and pandas' ``PerformanceWarning`` when many
+    rules (100+) are applied.
+
     Parameters
     ----------
     df
@@ -64,8 +70,10 @@ def apply_transformations(
         Copy of *df* with new columns added by the rules.
     """
     result = df.copy()
+    pending: dict[str, pd.Series[float]] = {}
     for rule in rules:
-        missing = [d for d in rule.dependencies if d not in result.columns]
+        available = set(result.columns) | set(pending)
+        missing = [d for d in rule.dependencies if d not in available]
         if missing:
             logger.warning(
                 "Skipping %s: missing dependencies %s",
@@ -73,5 +81,21 @@ def apply_transformations(
                 missing,
             )
             continue
-        result[rule.output_name] = rule.compute(result)
+        # rule.compute() reads from result; flush pending
+        # into result if this rule needs any pending column.
+        if any(d in pending for d in rule.dependencies):
+            result = pd.concat(
+                [
+                    result,
+                    pd.DataFrame(pending, index=result.index),
+                ],
+                axis=1,
+            )
+            pending = {}
+        pending[rule.output_name] = rule.compute(result)
+    if pending:
+        result = pd.concat(
+            [result, pd.DataFrame(pending, index=result.index)],
+            axis=1,
+        )
     return result

{tesorotools_python-0.0.26.dist-info → tesorotools_python-0.0.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tesorotools-python
-Version: 0.0.26
+Version: 0.0.28
 Requires-Python: >=3.13
 Requires-Dist: babel>=2.17
 Requires-Dist: eikon>=1.1

{tesorotools_python-0.0.26.dist-info → tesorotools_python-0.0.28.dist-info}/RECORD

@@ -6,7 +6,7 @@ tesorotools/artists/__init__.py,sha256=f9xOKEGJxQUE9nWwLft7cwtvPVmeyZRaF94oONKMZ
 tesorotools/artists/barh.md,sha256=VUYF4fnVrZo4BOET7l8uyMPAopEvoSMgfzBppF-AYug,4539
 tesorotools/artists/barh_plot.py,sha256=WcZl3rDdF-Jmf4uNYswpNn-JBPASwJWSEnCFWrlZFYk,16931
 tesorotools/artists/line_plot.py,sha256=_V7bKJFVQL-yMKDdhZX4hFizZs6FoW1wHmq7M_joY0Q,12992
-tesorotools/artists/stacked.py,sha256=skx92Ntjyk7DYlriwwXfPKStDNLYAhd0dnRKALJe6Ts,9563
+tesorotools/artists/stacked.py,sha256=YnsOm-l7hXcnBvfvaz43vUomoyplo_BF6IBa1bHELvU,10647
 tesorotools/artists/table.py,sha256=pyICq6yFEOSWGMc-3rhhgReodRImkBUg53kx8Msjf-c,8144
 tesorotools/artists/type_curve.py,sha256=raBdWt-2SpqujlIf7UQKAbKR1haln_zB7PJafUQmoLc,5963
 tesorotools/assets/README.md,sha256=pB77vD_MvE26ftVNcZuj3iS0y-6g-WzqHb5OScDkQ3I,289
@@ -36,7 +36,7 @@ tesorotools/offsets/offsets.py,sha256=-DOXy7gjI0r_UOxPtKv3OS10Z797dkeFMbjl8oWItP
 tesorotools/offsets/outliers.py,sha256=C2L-7d7jwx-Li7A73QO80lLzA4Mjp0mQBWbOqDppByM,677
 tesorotools/pipeline/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tesorotools/pipeline/diagnose.py,sha256=Ji_T3rAn-z4NhSdkpyPsnwQ1CTl4u5-1gnJZC8d1Sxg,1470
-tesorotools/pipeline/engine.py,sha256=R2sqpgUGRvNwMdTIn1HAMJzv0z_-XvqNgthDRKHrDEs,2030
+tesorotools/pipeline/engine.py,sha256=D4gKIv1bO6pw2gUMWJ7CVRv7e_kWWrWpZCzupeZCx0k,2929
 tesorotools/pipeline/rules.py,sha256=J5pevYdb8IixM87Gg6pSWNIpELVI770bjT6NnyvmGRw,5238
 tesorotools/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tesorotools/providers/base.py,sha256=IORHn8nHRcCDDnU_79q1Gc2tvuZ-X00f2yax24VX-f4,2201
@@ -59,6 +59,6 @@ tesorotools/utils/matplotlib.py,sha256=lItlkJwJEDGALt0J1UZioTz2GOOFPPe7ffQ4Hzxl7
 tesorotools/utils/series.py,sha256=AEf5DhneYjzQ0nvZD5b1IU3hop0Xgb3Bw2xWs_G3Lhw,1207
 tesorotools/utils/shortcuts.py,sha256=9sNrJHSNzCoZgBLWnG8U_dK6jTEjJUFgHPzUye7kU2U,860
 tesorotools/utils/template.py,sha256=jRT6ilDs87GuOmLltbCAP5-Lan7UeMQ_D_KuP-WwjIc,5343
-tesorotools_python-0.0.26.dist-info/METADATA,sha256=WRyRs9MvSy9NJDyuk9h6CVVGuvNv5UnzewrT8LyQ7b4,532
-tesorotools_python-0.0.26.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-tesorotools_python-0.0.26.dist-info/RECORD,,
+tesorotools_python-0.0.28.dist-info/METADATA,sha256=eXhh10WZle1az2cXgxz27wM9zCm0ax2ew-XtL3T_4Y0,532
+tesorotools_python-0.0.28.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+tesorotools_python-0.0.28.dist-info/RECORD,,