formative-ds 1.2.0__tar.gz → 1.2.2__tar.gz

{formative_ds-1.2.0 → formative_ds-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: formative-ds
-Version: 1.2.0
+Version: 1.2.2
 Summary: A Python library for quantitative reasoning.
 Project-URL: Homepage, https://getformative.dev
 Project-URL: Documentation, https://docs.getformative.dev
@@ -21,6 +21,7 @@ Classifier: Topic :: Scientific/Engineering :: Mathematics
 Requires-Python: >=3.10
 Requires-Dist: numpy<3,>=2
 Requires-Dist: pandas<4,>=2.1
+Requires-Dist: scipy<2,>=1.11
 Requires-Dist: statsmodels<0.15,>=0.14
 Description-Content-Type: text/markdown
 
@@ -47,7 +48,7 @@ Comprehensive documentation is available at [docs.getformative.dev](https://docs
 ### Causal estimation
 
 ```python
-from formative import DAG, OLSObservational
+from formative.causal import DAG, OLSObservational
 
 dag = DAG()
 dag.assume("ability").causes("education", "income")
@@ -119,7 +120,7 @@ To use `formative` from a script outside this repo without installing it, either
 import sys
 sys.path.insert(0, "/path/to/formative")
 
-from formative import DAG, OLSObservational
+from formative.causal import DAG, OLSObservational
 ```
 
 Or set `PYTHONPATH` before running:

{formative_ds-1.2.0 → formative_ds-1.2.2}/README.md

@@ -21,7 +21,7 @@ Comprehensive documentation is available at [docs.getformative.dev](https://docs
 ### Causal estimation
 
 ```python
-from formative import DAG, OLSObservational
+from formative.causal import DAG, OLSObservational
 
 dag = DAG()
 dag.assume("ability").causes("education", "income")
@@ -93,7 +93,7 @@ To use `formative` from a script outside this repo without installing it, either
 import sys
 sys.path.insert(0, "/path/to/formative")
 
-from formative import DAG, OLSObservational
+from formative.causal import DAG, OLSObservational
 ```
 
 Or set `PYTHONPATH` before running:

{formative_ds-1.2.0 → formative_ds-1.2.2}/docs/causal/estimators.rst

@@ -12,6 +12,7 @@ OLSResult
 
 .. autoclass:: formative.causal.OLSResult
    :members:
+   :inherited-members:
 
 IV2SLS
 ------
@@ -24,6 +25,7 @@ IVResult
 
 .. autoclass:: formative.causal.IVResult
    :members:
+   :inherited-members:
 
 PropensityScoreMatching
 -----------------------
@@ -36,6 +38,7 @@ MatchingResult
 
 .. autoclass:: formative.causal.MatchingResult
    :members:
+   :inherited-members:
 
 RCT
 ---
@@ -48,6 +51,7 @@ RCTResult
 
 .. autoclass:: formative.causal.RCTResult
    :members:
+   :inherited-members:
 
 DiD
 ---
@@ -60,6 +64,7 @@ DiDResult
 
 .. autoclass:: formative.causal.DiDResult
    :members:
+   :inherited-members:
 
 RDD
 ---
@@ -72,3 +77,4 @@ RDDResult
 
 .. autoclass:: formative.causal.RDDResult
    :members:
+   :inherited-members:

formative_ds-1.2.2/formative/causal/estimators/_base.py

@@ -0,0 +1,93 @@
+from __future__ import annotations
+
+from ..refutations._check import Assumption
+
+
+class _BaseResult:
+    """
+    Shared behaviour for estimation result classes.
+
+    Subclasses set ``self._treatment`` and ``self._outcome`` in ``__init__``
+    and define a class-level ``_ASSUMPTIONS`` list. ``decide()`` reads the
+    ``effect`` / ``std_err`` / ``conf_int`` properties, which subclasses
+    (or ``_StatsmodelsResult``) provide.
+    """
+
+    _ASSUMPTIONS: list[Assumption] = []
+    _treatment: str
+    _outcome: str
+
+    @property
+    def assumptions(self) -> list[Assumption]:
+        """Modelling assumptions required for a causal interpretation."""
+        return list(self._ASSUMPTIONS)
+
+    def _assumptions_lines(self, width: int = 48) -> list[str]:
+        """Assumptions footer appended to every ``summary()``."""
+        lines = ["", "  Assumptions", "  " + "┄" * width]
+        lines += [f"  {a.fmt_tag()}  {a.name}" for a in self._ASSUMPTIONS]
+        lines.append("")
+        return lines
+
+    def decide(self, cost: float, benefit: float):
+        """
+        Compute a cost-benefit decision analysis from this causal estimate.
+
+        Parameters
+        ----------
+        cost : float
+            Cost per unit of treatment applied.
+        benefit : float
+            Benefit (revenue, utility, etc.) per unit increase in the outcome.
+
+        Returns
+        -------
+        DecisionReport
+            Optimal decision, net benefit, CI, confidence, and robustness flag.
+        """
+        from ..decision import _decide
+
+        return _decide(self.effect, self.std_err, self.conf_int, self._treatment, self._outcome, cost, benefit)
+
+    def __repr__(self) -> str:
+        return self.summary()
+
+
+class _StatsmodelsResult(_BaseResult):
+    """
+    Result backed by a fitted statsmodels regression.
+
+    Effect statistics are read from ``self._result`` at parameter
+    ``self._param``, which defaults to the treatment name (DiD overrides
+    it with the group × time interaction term).
+    """
+
+    @property
+    def _param(self) -> str:
+        return self._treatment
+
+    @property
+    def effect(self) -> float:
+        """Point estimate of the causal effect of treatment on outcome."""
+        return float(self._result.params[self._param])
+
+    @property
+    def std_err(self) -> float:
+        """Standard error of the treatment effect estimate."""
+        return float(self._result.bse[self._param])
+
+    @property
+    def conf_int(self) -> tuple[float, float]:
+        """95% confidence interval for the treatment effect."""
+        ci = self._result.conf_int()
+        return (float(ci.loc[self._param, 0]), float(ci.loc[self._param, 1]))
+
+    @property
+    def pvalue(self) -> float:
+        """p-value for the treatment effect (``H0: effect = 0``)."""
+        return float(self._result.pvalues[self._param])
+
+    @property
+    def statsmodels_result(self):
+        """The underlying statsmodels result, for full diagnostics."""
+        return self._result

{formative_ds-1.2.0 → formative_ds-1.2.2}/formative/causal/estimators/did.py

@@ -5,6 +5,7 @@ import statsmodels.formula.api as smf
 
 from ..dag import DAG
 from ..refutations._check import Assumption
+from ._base import _StatsmodelsResult
 
 DID_ASSUMPTIONS: list[Assumption] = [
     Assumption(
@@ -17,7 +18,7 @@ DID_ASSUMPTIONS: list[Assumption] = [
 ]
 
 
-class DiDResult:
+class DiDResult(_StatsmodelsResult):
     """
     The result of a Difference-in-Differences estimation.
 
@@ -26,6 +27,8 @@ class DiDResult:
     control group's trajectory.
     """
 
+    _ASSUMPTIONS = DID_ASSUMPTIONS
+
     def __init__(
         self,
         result,
@@ -45,45 +48,20 @@ class DiDResult:
         self._n = n
 
     @property
-    def _interaction(self) -> str:
+    def _param(self) -> str:
+        # The DiD estimate is the group × time interaction coefficient.
         return f"{self._group}:{self._time}"
 
     @property
-    def effect(self) -> float:
-        """DiD point estimate: ATT under the parallel trends assumption."""
-        return float(self._result.params[self._interaction])
+    def _treatment(self) -> str:
+        # The treatment label used by decide() is the group indicator.
+        return self._group
 
     @property
     def naive_diff(self) -> float:
         """Naive post-period difference: treated mean minus control mean in the post period."""
         return self._naive_diff
 
-    @property
-    def std_err(self) -> float:
-        """Standard error of the DiD estimate."""
-        return float(self._result.bse[self._interaction])
-
-    @property
-    def conf_int(self) -> tuple[float, float]:
-        """95% confidence interval for the DiD estimate."""
-        ci = self._result.conf_int()
-        return (float(ci.loc[self._interaction, 0]), float(ci.loc[self._interaction, 1]))
-
-    @property
-    def pvalue(self) -> float:
-        """p-value for the DiD estimate (``H0: ATT = 0``)."""
-        return float(self._result.pvalues[self._interaction])
-
-    @property
-    def statsmodels_result(self):
-        """The underlying statsmodels OLS result, for full diagnostics."""
-        return self._result
-
-    @property
-    def assumptions(self) -> list[Assumption]:
-        """Modelling assumptions required for a causal interpretation."""
-        return list(DID_ASSUMPTIONS)
-
     def executive_summary(self) -> str:
         """Narrative explanation of the method, DAG, assumptions, and result."""
         from .._explain import explain_did
@@ -107,35 +85,10 @@ class DiDResult:
             f"  95% CI               : [{lo:.4f}, {hi:.4f}]",
             f"  p-value              : {self.pvalue:>10.4f}",
             f"  N                    : {self._n:>10}",
-            "",
-            "  Assumptions",
-            "  " + "\u2504" * 48,
         ]
-        for a in DID_ASSUMPTIONS:
-            lines.append(f"  {a.fmt_tag()}  {a.name}")
-        lines.append("")
+        lines += self._assumptions_lines()
         return "\n".join(lines)
 
-    def decide(self, cost: float, benefit: float):
-        """
-        Compute a cost-benefit decision analysis from this causal estimate.
-
-        Parameters
-        ----------
-        cost : float
-            Cost per unit of treatment applied.
-        benefit : float
-            Benefit (revenue, utility, etc.) per unit increase in the outcome.
-
-        Returns
-        -------
-        DecisionReport
-            Optimal decision, net benefit, CI, confidence, and robustness flag.
-        """
-        from ..decision import _decide
-
-        return _decide(self.effect, self.std_err, self.conf_int, self._group, self._outcome, cost, benefit)
-
     def refute(self, data: pd.DataFrame):
         """
         Run refutation checks against this DiD estimation.

{formative_ds-1.2.0 → formative_ds-1.2.2}/formative/causal/estimators/iv.py

@@ -7,6 +7,7 @@ from statsmodels.sandbox.regression.gmm import IV2SLS as _IV2SLS
 
 from ..dag import DAG
 from ..refutations._check import Assumption
+from ._base import _StatsmodelsResult
 
 IV_ASSUMPTIONS: list[Assumption] = [
     Assumption("Relevance: the instrument strongly affects treatment", testable=True),
@@ -17,7 +18,7 @@ IV_ASSUMPTIONS: list[Assumption] = [
 ]
 
 
-class IVResult:
+class IVResult(_StatsmodelsResult):
     """
     The result of an IV (2SLS) causal estimation.
 
@@ -26,6 +27,8 @@ class IVResult:
     bias that IV corrects.
     """
 
+    _ASSUMPTIONS = IV_ASSUMPTIONS
+
     def __init__(
         self,
         result,
@@ -46,78 +49,27 @@ class IVResult:
         self._dag = dag
         self._n = n
 
-    @property
-    def effect(self) -> float:
-        """2SLS point estimate of the causal effect of treatment on outcome."""
-        return float(self._result.params[self._treatment])
-
     @property
     def unadjusted_effect(self) -> float:
         """Unadjusted OLS estimate: naive regression without instrument or controls."""
         return float(self._unadjusted.params[self._treatment])
 
-    @property
-    def std_err(self) -> float:
-        """Standard error of the 2SLS treatment effect estimate."""
-        return float(self._result.bse[self._treatment])
-
-    @property
-    def conf_int(self) -> tuple[float, float]:
-        """95% confidence interval for the 2SLS treatment effect."""
-        ci = self._result.conf_int()
-        return (float(ci.loc[self._treatment, 0]), float(ci.loc[self._treatment, 1]))
-
-    @property
-    def pvalue(self) -> float:
-        """p-value for the 2SLS treatment effect (``H0: effect = 0``)."""
-        return float(self._result.pvalues[self._treatment])
-
     @property
     def adjustment_set(self) -> set[str]:
         """Observed confounders included as controls in both OLS stages."""
         return self._adjustment_set
 
-    @property
-    def statsmodels_result(self):
-        """The underlying statsmodels IV2SLS result, for full diagnostics."""
-        return self._result
-
     @property
     def statsmodels_unadjusted_result(self):
         """The underlying unadjusted OLS result, for full diagnostics."""
         return self._unadjusted
 
-    @property
-    def assumptions(self) -> list[Assumption]:
-        """Modelling assumptions required for a causal interpretation."""
-        return list(IV_ASSUMPTIONS)
-
     def executive_summary(self) -> str:
         """Narrative explanation of the method, DAG, assumptions, and result."""
         from .._explain import explain_iv
 
         return explain_iv(self)
 
-    def decide(self, cost: float, benefit: float):
-        """
-        Compute a cost-benefit decision analysis from this causal estimate.
-
-        Parameters
-        ----------
-        cost : float
-            Cost per unit of treatment applied.
-        benefit : float
-            Benefit (revenue, utility, etc.) per unit increase in the outcome.
-
-        Returns
-        -------
-        DecisionReport
-            Optimal decision, net benefit, CI, confidence, and robustness flag.
-        """
-        from ..decision import _decide
-
-        return _decide(self.effect, self.std_err, self.conf_int, self._treatment, self._outcome, cost, benefit)
-
     def refute(self, data: pd.DataFrame):
         """
         Run refutation checks against this IV estimation.
@@ -181,18 +133,10 @@ class IVResult:
             f"  95% CI               : [{lo:.4f}, {hi:.4f}]",
             f"  p-value              : {self.pvalue:>10.4f}",
             f"  N                    : {self._n:>10}",
-            "",
-            "  Assumptions",
-            "  " + "┄" * 48,
         ]
-        for a in IV_ASSUMPTIONS:
-            lines.append(f"  {a.fmt_tag()}  {a.name}")
-        lines.append("")
+        lines += self._assumptions_lines()
         return "\n".join(lines)
 
-    def __repr__(self) -> str:
-        return self.summary()
-
 
 class IV2SLS:
     """

{formative_ds-1.2.0 → formative_ds-1.2.2}/formative/causal/estimators/matching.py

@@ -7,6 +7,7 @@ import statsmodels.formula.api as smf
 from .._exceptions import IdentificationError
 from ..dag import DAG
 from ..refutations._check import Assumption
+from ._base import _BaseResult
 
 MATCHING_ASSUMPTIONS: list[Assumption] = [
     Assumption("Conditional independence: no unobserved confounders given matched variables", testable=False),
@@ -65,7 +66,7 @@ def _att_from_ps(
 # ── Result ─────────────────────────────────────────────────────────────────────
 
 
-class MatchingResult:
+class MatchingResult(_BaseResult):
     """
     The result of a propensity score matching estimation.
 
@@ -75,6 +76,8 @@ class MatchingResult:
     full matching procedure.
     """
 
+    _ASSUMPTIONS = MATCHING_ASSUMPTIONS
+
     def __init__(
         self,
         att: float,
@@ -136,11 +139,6 @@ class MatchingResult:
         """Full array of per-bootstrap ATT values, for diagnostics."""
         return self._bootstrap_atts.copy()
 
-    @property
-    def assumptions(self) -> list[Assumption]:
-        """Modelling assumptions required for a causal interpretation."""
-        return list(MATCHING_ASSUMPTIONS)
-
     def executive_summary(self) -> str:
         """Narrative explanation of the method, DAG, assumptions, and result."""
         from .._explain import explain_matching
@@ -179,35 +177,10 @@ class MatchingResult:
             f"  N                    : {self._n:>10}",
             "",
             "  Matching: 1-to-1 nearest-neighbour on propensity score (with replacement)",
-            "",
-            "  Assumptions",
-            "  " + "┄" * 48,
         ]
-        for a in MATCHING_ASSUMPTIONS:
-            lines.append(f"  {a.fmt_tag()}  {a.name}")
-        lines.append("")
+        lines += self._assumptions_lines()
         return "\n".join(lines)
 
-    def decide(self, cost: float, benefit: float):
-        """
-        Compute a cost-benefit decision analysis from this causal estimate.
-
-        Parameters
-        ----------
-        cost : float
-            Cost per unit of treatment applied.
-        benefit : float
-            Benefit (revenue, utility, etc.) per unit increase in the outcome.
-
-        Returns
-        -------
-        DecisionReport
-            Optimal decision, net benefit, CI, confidence, and robustness flag.
-        """
-        from ..decision import _decide
-
-        return _decide(self.effect, self.std_err, self.conf_int, self._treatment, self._outcome, cost, benefit)
-
     def refute(self, data: pd.DataFrame):
         """
         Run refutation checks against this matching estimation.
@@ -254,9 +227,6 @@ class MatchingResult:
             outcome=self._outcome,
         )
 
-    def __repr__(self) -> str:
-        return self.summary()
-
 
 # ── Estimator ──────────────────────────────────────────────────────────────────
 

{formative_ds-1.2.0 → formative_ds-1.2.2}/formative/causal/estimators/ols.py

@@ -6,6 +6,7 @@ import statsmodels.formula.api as smf
 from .._exceptions import IdentificationError
 from ..dag import DAG
 from ..refutations._check import Assumption
+from ._base import _StatsmodelsResult
 
 OLS_ASSUMPTIONS: list[Assumption] = [
     Assumption("No unobserved confounders (selection on observables)", testable=False),
@@ -16,7 +17,7 @@ OLS_ASSUMPTIONS: list[Assumption] = [
 ]
 
 
-class OLSResult:
+class OLSResult(_StatsmodelsResult):
     """
     The result of an OLS causal estimation.
 
@@ -25,6 +26,8 @@ class OLSResult:
     effect of controlling for confounders directly.
     """
 
+    _ASSUMPTIONS = OLS_ASSUMPTIONS
+
     def __init__(
         self,
         adjusted_result,
@@ -35,7 +38,7 @@ class OLSResult:
         dag,
         n: int,
     ) -> None:
-        self._adjusted = adjusted_result
+        self._result = adjusted_result
         self._unadjusted = unadjusted_result
         self._treatment = treatment
         self._outcome = outcome
@@ -43,52 +46,21 @@ class OLSResult:
         self._dag = dag
         self._n = n
 
-    @property
-    def effect(self) -> float:
-        """Adjusted point estimate: causal effect of treatment on outcome."""
-        return float(self._adjusted.params[self._treatment])
-
     @property
     def unadjusted_effect(self) -> float:
         """Unadjusted point estimate: naive regression without controlling for confounders."""
         return float(self._unadjusted.params[self._treatment])
 
-    @property
-    def std_err(self) -> float:
-        """Standard error of the adjusted treatment effect estimate."""
-        return float(self._adjusted.bse[self._treatment])
-
-    @property
-    def conf_int(self) -> tuple[float, float]:
-        """95% confidence interval for the adjusted treatment effect."""
-        ci = self._adjusted.conf_int()
-        return (float(ci.loc[self._treatment, 0]), float(ci.loc[self._treatment, 1]))
-
-    @property
-    def pvalue(self) -> float:
-        """p-value for the adjusted treatment effect (``H0: effect = 0``)."""
-        return float(self._adjusted.pvalues[self._treatment])
-
     @property
     def adjustment_set(self) -> set[str]:
         """Variables controlled for to satisfy the backdoor criterion."""
         return self._adjustment_set
 
-    @property
-    def statsmodels_result(self):
-        """The underlying adjusted statsmodels result, for full diagnostics."""
-        return self._adjusted
-
     @property
     def statsmodels_unadjusted_result(self):
         """The underlying unadjusted statsmodels result, for full diagnostics."""
         return self._unadjusted
 
-    @property
-    def assumptions(self) -> list[Assumption]:
-        """Modelling assumptions required for a causal interpretation."""
-        return list(OLS_ASSUMPTIONS)
-
     def executive_summary(self) -> str:
         """Narrative explanation of the method, DAG, assumptions, and result."""
         from .._explain import explain_ols
@@ -124,35 +96,10 @@ class OLSResult:
             f"  95% CI               : [{lo:.4f}, {hi:.4f}]",
             f"  p-value              : {self.pvalue:>10.4f}",
             f"  N                    : {self._n:>10}",
-            "",
-            "  Assumptions",
-            "  " + "┄" * 48,
         ]
-        for a in OLS_ASSUMPTIONS:
-            lines.append(f"  {a.fmt_tag()}  {a.name}")
-        lines.append("")
+        lines += self._assumptions_lines()
         return "\n".join(lines)
 
-    def decide(self, cost: float, benefit: float):
-        """
-        Compute a cost-benefit decision analysis from this causal estimate.
-
-        Parameters
-        ----------
-        cost : float
-            Cost per unit of treatment applied.
-        benefit : float
-            Benefit (revenue, utility, etc.) per unit increase in the outcome.
-
-        Returns
-        -------
-        DecisionReport
-            Optimal decision, net benefit, CI, confidence, and robustness flag.
-        """
-        from ..decision import _decide
-
-        return _decide(self.effect, self.std_err, self.conf_int, self._treatment, self._outcome, cost, benefit)
-
     def refute(self, data: pd.DataFrame):
         """
         Run refutation checks against this OLS estimation.
@@ -186,9 +133,6 @@ class OLSResult:
             outcome=self._outcome,
         )
 
-    def __repr__(self) -> str:
-        return self.summary()
-
 
 class OLSObservational:
     """

{formative_ds-1.2.0 → formative_ds-1.2.2}/formative/causal/estimators/rct.py

@@ -5,6 +5,7 @@ import statsmodels.formula.api as smf
 
 from ..dag import DAG
 from ..refutations._check import Assumption
+from ._base import _StatsmodelsResult
 
 RCT_ASSUMPTIONS: list[Assumption] = [
     Assumption("Random assignment of treatment", testable=False),
@@ -13,7 +14,7 @@ RCT_ASSUMPTIONS: list[Assumption] = [
 ]
 
 
-class RCTResult:
+class RCTResult(_StatsmodelsResult):
     """
     The result of an RCT causal estimation.
 
@@ -22,6 +23,8 @@ class RCTResult:
     equals the difference in mean outcomes between treatment and control.
     """
 
+    _ASSUMPTIONS = RCT_ASSUMPTIONS
+
     def __init__(
         self,
         result,
@@ -36,37 +39,6 @@ class RCTResult:
         self._dag = dag
         self._n = n
 
-    @property
-    def effect(self) -> float:
-        """ATE: average treatment effect (difference in means)."""
-        return float(self._result.params[self._treatment])
-
-    @property
-    def std_err(self) -> float:
-        """Standard error of the ATE estimate."""
-        return float(self._result.bse[self._treatment])
-
-    @property
-    def conf_int(self) -> tuple[float, float]:
-        """95% confidence interval for the ATE."""
-        ci = self._result.conf_int()
-        return (float(ci.loc[self._treatment, 0]), float(ci.loc[self._treatment, 1]))
-
-    @property
-    def pvalue(self) -> float:
-        """p-value for the ATE (``H0: ATE = 0``)."""
-        return float(self._result.pvalues[self._treatment])
-
-    @property
-    def statsmodels_result(self):
-        """The underlying statsmodels OLS result, for full diagnostics."""
-        return self._result
-
-    @property
-    def assumptions(self) -> list[Assumption]:
-        """Modelling assumptions required for a causal interpretation."""
-        return list(RCT_ASSUMPTIONS)
-
     def executive_summary(self) -> str:
         """Narrative explanation of the method, DAG, assumptions, and result."""
         from .._explain import explain_rct
@@ -87,35 +59,10 @@ class RCTResult:
             f"  95% CI               : [{lo:.4f}, {hi:.4f}]",
             f"  p-value              : {self.pvalue:>10.4f}",
             f"  N                    : {self._n:>10}",
-            "",
-            "  Assumptions",
-            "  " + "┄" * 48,
         ]
-        for a in RCT_ASSUMPTIONS:
-            lines.append(f"  {a.fmt_tag()}  {a.name}")
-        lines.append("")
+        lines += self._assumptions_lines()
         return "\n".join(lines)
 
-    def decide(self, cost: float, benefit: float):
-        """
-        Compute a cost-benefit decision analysis from this causal estimate.
-
-        Parameters
-        ----------
-        cost : float
-            Cost per unit of treatment applied.
-        benefit : float
-            Benefit (revenue, utility, etc.) per unit increase in the outcome.
-
-        Returns
-        -------
-        DecisionReport
-            Optimal decision, net benefit, CI, confidence, and robustness flag.
-        """
-        from ..decision import _decide
-
-        return _decide(self.effect, self.std_err, self.conf_int, self._treatment, self._outcome, cost, benefit)
-
     def refute(self, data: pd.DataFrame):
         """
         Run refutation checks against this RCT estimation.
@@ -149,9 +96,6 @@ class RCTResult:
             outcome=self._outcome,
         )
 
-    def __repr__(self) -> str:
-        return self.summary()
-
 
 class RCT:
     """

{formative_ds-1.2.0 → formative_ds-1.2.2}/formative/causal/estimators/rdd.py

@@ -5,6 +5,7 @@ import statsmodels.formula.api as smf
 
 from ..dag import DAG
 from ..refutations._check import Assumption
+from ._base import _StatsmodelsResult
 
 RDD_ASSUMPTIONS: list[Assumption] = [
     Assumption(
@@ -23,7 +24,7 @@ RDD_ASSUMPTIONS: list[Assumption] = [
 ]
 
 
-class RDDResult:
+class RDDResult(_StatsmodelsResult):
     """
     The result of a Regression Discontinuity Design estimation.
 
@@ -31,6 +32,8 @@ class RDDResult:
     the cutoff, estimated via local linear regression on both sides of the threshold.
     """
 
+    _ASSUMPTIONS = RDD_ASSUMPTIONS
+
     def __init__(
         self,
         result,
@@ -51,42 +54,11 @@ class RDDResult:
         self._unadjusted_effect = unadjusted_effect
         self._dag = dag
 
-    @property
-    def effect(self) -> float:
-        """LATE at the cutoff: the jump in outcome at the threshold (coefficient on the treatment indicator)."""
-        return float(self._result.params[self._treatment])
-
     @property
     def unadjusted_effect(self) -> float:
         """Naive mean difference: above-cutoff mean minus below-cutoff mean."""
         return self._unadjusted_effect
 
-    @property
-    def std_err(self) -> float:
-        """Standard error of the LATE at the cutoff estimate."""
-        return float(self._result.bse[self._treatment])
-
-    @property
-    def conf_int(self) -> tuple[float, float]:
-        """95% confidence interval for the LATE at the cutoff estimate."""
-        ci = self._result.conf_int()
-        return (float(ci.loc[self._treatment, 0]), float(ci.loc[self._treatment, 1]))
-
-    @property
-    def pvalue(self) -> float:
-        """p-value for the LATE at the cutoff (``H0: LATE at cutoff = 0``)."""
-        return float(self._result.pvalues[self._treatment])
-
-    @property
-    def statsmodels_result(self):
-        """The underlying statsmodels OLS result, for full diagnostics."""
-        return self._result
-
-    @property
-    def assumptions(self) -> list[Assumption]:
-        """Modelling assumptions required for a causal interpretation."""
-        return list(RDD_ASSUMPTIONS)
-
     @property
     def cutoff(self) -> float:
         """The threshold value that determines treatment assignment."""
@@ -131,35 +103,10 @@ class RDDResult:
             f"  95% CI                : [{lo:.4f}, {hi:.4f}]",
             f"  p-value               : {self.pvalue:>10.4f}",
             f"  N (in bandwidth)      : {self.n_obs:>9}",
-            "",
-            "  Assumptions",
-            "  " + "\u2504" * 30,
         ]
-        for a in RDD_ASSUMPTIONS:
-            lines.append(f"  {a.fmt_tag()}  {a.name}")
-        lines.append("")
+        lines += self._assumptions_lines(width=30)
         return "\n".join(lines)
 
-    def decide(self, cost: float, benefit: float):
-        """
-        Compute a cost-benefit decision analysis from this causal estimate.
-
-        Parameters
-        ----------
-        cost : float
-            Cost per unit of treatment applied.
-        benefit : float
-            Benefit (revenue, utility, etc.) per unit increase in the outcome.
-
-        Returns
-        -------
-        DecisionReport
-            Optimal decision, net benefit, CI, confidence, and robustness flag.
-        """
-        from ..decision import _decide
-
-        return _decide(self.effect, self.std_err, self.conf_int, self._treatment, self._outcome, cost, benefit)
-
     def refute(self, data: pd.DataFrame):
         """
         Run refutation checks against this RDD estimation.
@@ -210,9 +157,6 @@ class RDDResult:
             outcome=self._outcome,
         )
 
-    def __repr__(self) -> str:
-        return self.summary()
-
 
 class RDD:
     """

{formative_ds-1.2.0 → formative_ds-1.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "formative-ds"
-version = "1.2.0"
+version = "1.2.2"
 description = "A Python library for quantitative reasoning."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -25,6 +25,7 @@ classifiers = [
 
 dependencies = [
     "numpy>=2,<3",
+    "scipy>=1.11,<2",
     "statsmodels>=0.14,<0.15",
     "pandas>=2.1,<4",
 ]
@@ -42,7 +43,7 @@ docs = ["sphinx>=7,<10", "sphinx-rtd-theme>=2,<4", "sphinx-copybutton>=0.5,<0.6"
 packages = ["formative"]
 
 [tool.bumpversion]
-current_version = "1.2.0"
+current_version = "1.2.2"
 commit = true
 tag = true
 tag_name = "v{new_version}"

{formative_ds-1.2.0 → formative_ds-1.2.2}/uv.lock

@@ -303,13 +303,15 @@ wheels = [
 
 [[package]]
 name = "formative-ds"
-version = "1.1.1"
+version = "1.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "numpy", version = "2.2.6", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.11'" },
     { name = "numpy", version = "2.4.2", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.11'" },
     { name = "pandas", version = "2.3.3", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.11'" },
     { name = "pandas", version = "3.0.1", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.11'" },
+    { name = "scipy", version = "1.15.3", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.11'" },
+    { name = "scipy", version = "1.17.1", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.11'" },
     { name = "statsmodels" },
 ]
 
@@ -334,6 +336,7 @@ docs = [
 requires-dist = [
     { name = "numpy", specifier = ">=2,<3" },
     { name = "pandas", specifier = ">=2.1,<4" },
+    { name = "scipy", specifier = ">=1.11,<2" },
     { name = "statsmodels", specifier = ">=0.14,<0.15" },
 ]
 
@@ -352,11 +355,11 @@ docs = [
 
 [[package]]
 name = "idna"
-version = "3.11"
+version = "3.15"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/6f/6d/0703ccc57f3a7233505399edb88de3cbd678da106337b9fcde432b65ed60/idna-3.11.tar.gz", hash = "sha256:795dafcc9c04ed0c1fb032c2aa73654d8e8c5023a7df64a53f39190ada629902", size = 194582, upload-time = "2025-10-12T14:55:20.501Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/82/77/7b3966d0b9d1d31a36ddf1746926a11dface89a83409bf1483f0237aa758/idna-3.15.tar.gz", hash = "sha256:ca962446ea538f7092a95e057da437618e886f4d349216d2b1e294abfdb65fdc", size = 199245, upload-time = "2026-05-12T22:45:57.011Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/0e/61/66938bbb5fc52dbdf84594873d5b51fb1f7c7794e9c0f5bd885f30bc507b/idna-3.11-py3-none-any.whl", hash = "sha256:771a87f49d9defaf64091e6e6fe9c18d4833f140bd19464795bc32d966ca37ea", size = 71008, upload-time = "2025-10-12T14:55:18.883Z" },
+    { url = "https://files.pythonhosted.org/packages/d2/23/408243171aa9aaba178d3e2559159c24c1171a641aa83b67bdd3394ead8e/idna-3.15-py3-none-any.whl", hash = "sha256:048adeaf8c2d788c40fee287673ccaa74c24ffd8dcf09ffa555a2fbb59f10ac8", size = 72340, upload-time = "2026-05-12T22:45:55.733Z" },
 ]
 
 [[package]]
@@ -797,16 +800,16 @@ wheels = [
 
 [[package]]
 name = "pygments"
-version = "2.19.2"
+version = "2.20.0"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/b0/77/a5b8c569bf593b0140bde72ea885a803b82086995367bf2037de0159d924/pygments-2.19.2.tar.gz", hash = "sha256:636cb2477cec7f8952536970bc533bc43743542f70392ae026374600add5b887", size = 4968631, upload-time = "2025-06-21T13:39:12.283Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/c3/b2/bc9c9196916376152d655522fdcebac55e66de6603a76a02bca1b6414f6c/pygments-2.20.0.tar.gz", hash = "sha256:6757cd03768053ff99f3039c1a36d6c0aa0b263438fcab17520b30a303a82b5f", size = 4955991, upload-time = "2026-03-29T13:29:33.898Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/c7/21/705964c7812476f378728bdf590ca4b771ec72385c533964653c68e86bdc/pygments-2.19.2-py3-none-any.whl", hash = "sha256:86540386c03d588bb81d44bc3928634ff26449851e99741617ecb9037ee5ec0b", size = 1225217, upload-time = "2025-06-21T13:39:07.939Z" },
+    { url = "https://files.pythonhosted.org/packages/f4/7e/a72dd26f3b0f4f2bf1dd8923c85f7ceb43172af56d63c7383eb62b332364/pygments-2.20.0-py3-none-any.whl", hash = "sha256:81a9e26dd42fd28a23a2d169d86d7ac03b46e2f8b59ed4698fb4785f946d0176", size = 1231151, upload-time = "2026-03-29T13:29:30.038Z" },
 ]
 
 [[package]]
 name = "pytest"
-version = "9.0.2"
+version = "9.0.3"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "colorama", marker = "sys_platform == 'win32'" },
@@ -817,9 +820,9 @@ dependencies = [
     { name = "pygments" },
     { name = "tomli", marker = "python_full_version < '3.11'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/d1/db/7ef3487e0fb0049ddb5ce41d3a49c235bf9ad299b6a25d5780a89f19230f/pytest-9.0.2.tar.gz", hash = "sha256:75186651a92bd89611d1d9fc20f0b4345fd827c41ccd5c299a868a05d70edf11", size = 1568901, upload-time = "2025-12-06T21:30:51.014Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/7d/0d/549bd94f1a0a402dc8cf64563a117c0f3765662e2e668477624baeec44d5/pytest-9.0.3.tar.gz", hash = "sha256:b86ada508af81d19edeb213c681b1d48246c1a91d304c6c81a427674c17eb91c", size = 1572165, upload-time = "2026-04-07T17:16:18.027Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/3b/ab/b3226f0bd7cdcf710fbede2b3548584366da3b19b5021e74f5bde2a8fa3f/pytest-9.0.2-py3-none-any.whl", hash = "sha256:711ffd45bf766d5264d487b917733b453d917afd2b0ad65223959f59089f875b", size = 374801, upload-time = "2025-12-06T21:30:49.154Z" },
+    { url = "https://files.pythonhosted.org/packages/d4/24/a372aaf5c9b7208e7112038812994107bc65a84cd00e0354a88c2c77a617/pytest-9.0.3-py3-none-any.whl", hash = "sha256:2c5efc453d45394fdd706ade797c0a81091eccd1d6e4bccfcd476e2b8e0ab5d9", size = 375249, upload-time = "2026-04-07T17:16:16.13Z" },
 ]
 
 [[package]]
@@ -1438,9 +1441,9 @@ wheels = [
 
 [[package]]
 name = "urllib3"
-version = "2.6.3"
+version = "2.7.0"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/c7/24/5f1b3bdffd70275f6661c76461e25f024d5a38a46f04aaca912426a2b1d3/urllib3-2.6.3.tar.gz", hash = "sha256:1b62b6884944a57dbe321509ab94fd4d3b307075e0c2eae991ac71ee15ad38ed", size = 435556, upload-time = "2026-01-07T16:24:43.925Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/53/0c/06f8b233b8fd13b9e5ee11424ef85419ba0d8ba0b3138bf360be2ff56953/urllib3-2.7.0.tar.gz", hash = "sha256:231e0ec3b63ceb14667c67be60f2f2c40a518cb38b03af60abc813da26505f4c", size = 433602, upload-time = "2026-05-07T16:13:18.596Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/39/08/aaaad47bc4e9dc8c725e68f9d04865dbcb2052843ff09c97b08904852d84/urllib3-2.6.3-py3-none-any.whl", hash = "sha256:bf272323e553dfb2e87d9bfd225ca7b0f467b919d7bbd355436d3fd37cb0acd4", size = 131584, upload-time = "2026-01-07T16:24:42.685Z" },
+    { url = "https://files.pythonhosted.org/packages/7f/3e/5db95bcf282c52709639744ca2a8b149baccf648e39c8cc87553df9eae0c/urllib3-2.7.0-py3-none-any.whl", hash = "sha256:9fb4c81ebbb1ce9531cce37674bbc6f1360472bc18ca9a553ede278ef7276897", size = 131087, upload-time = "2026-05-07T16:13:17.151Z" },
 ]