Generalize calibration bins

src/rtichoke/_calibration_viz_spec_v2.py

@@ -35,7 +35,9 @@ def _calibration_v2_spec_from_curve_list(
             f"Unsupported calibration type {calibration_type!r}; expected {supported}."
         )
 
-    point_key = "deciles_dat" if calibration_type == "discrete" else "smooth_dat"
+    point_key = (
+        "calibration_bins_dat" if calibration_type == "discrete" else "smooth_dat"
+    )
     point_frame = _require_frame(calibration_curve_list, point_key)
     distribution_frame = _require_frame(
         calibration_curve_list, "histogram_for_calibration"

src/rtichoke/_viz_browser.py

@@ -67,7 +67,7 @@ def _calibration_spec_from_curve_list(
     calibration_curve_list: dict[str, Any],
 ) -> dict[str, object]:
     """Map existing discrete calibration output to the canonical spec."""
-    rows = calibration_curve_list["deciles_dat"].select(
+    rows = calibration_curve_list["calibration_bins_dat"].select(
         "reference_group", "x", "y", "n_reals", "n"
     )
     distribution_rows = calibration_curve_list["histogram_for_calibration"].select(

src/rtichoke/calibration/__init__.py

@@ -2,6 +2,7 @@
 Subpackage for Calibration
 """
 
+import functools
 import numpy as np
 
 from . import calibration as _calibration
@@ -38,6 +39,7 @@ def _validate_outcome_values(reals, allowed_values):
             raise ValueError("Time-dependent outcomes must contain only 0, 1, and 2.")
 
 
+@functools.wraps(_original_create_calibration_curve)
 def create_calibration_curve(*args, **kwargs):
     """Create an interactive calibration plot with a square main panel."""
     probs = _argument(args, kwargs, "probs", 0)
@@ -50,6 +52,7 @@ def create_calibration_curve(*args, **kwargs):
     )
 
 
+@functools.wraps(_original_create_calibration_curve_times)
 def create_calibration_curve_times(*args, **kwargs):
     """Create an interactive time-dependent calibration plot with a square main panel."""
     probs = _argument(args, kwargs, "probs", 0)