diff-diff 0.5.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

diff_diff/__init__.py

@@ -5,16 +5,42 @@ This library provides sklearn-like estimators for causal inference
 using the difference-in-differences methodology.
 """
 
+from diff_diff.diagnostics import (
+    PlaceboTestResults,
+    leave_one_out_test,
+    permutation_test,
+    placebo_group_test,
+    placebo_timing_test,
+    run_all_placebo_tests,
+    run_placebo_test,
+)
 from diff_diff.estimators import (
     DifferenceInDifferences,
-    TwoWayFixedEffects,
     MultiPeriodDiD,
     SyntheticDiD,
+    TwoWayFixedEffects,
 )
-from diff_diff.staggered import (
-    CallawaySantAnna,
-    CallawaySantAnnaResults,
-    GroupTimeEffect,
+from diff_diff.honest_did import (
+    DeltaRM,
+    DeltaSD,
+    DeltaSDRM,
+    HonestDiD,
+    HonestDiDResults,
+    SensitivityResults,
+    compute_honest_did,
+    sensitivity_plot,
+)
+from diff_diff.prep import (
+    aggregate_to_cohorts,
+    balance_panel,
+    create_event_time,
+    generate_did_data,
+    make_post_indicator,
+    make_treatment_indicator,
+    rank_control_units,
+    summarize_did_data,
+    validate_did_data,
+    wide_to_long,
 )
 from diff_diff.results import (
     DiDResults,
@@ -22,40 +48,26 @@ from diff_diff.results import (
     PeriodEffect,
     SyntheticDiDResults,
 )
-from diff_diff.visualization import (
-    plot_event_study,
-    plot_group_effects,
-)
-from diff_diff.prep import (
-    make_treatment_indicator,
-    make_post_indicator,
-    wide_to_long,
-    balance_panel,
-    validate_did_data,
-    summarize_did_data,
-    generate_did_data,
-    create_event_time,
-    aggregate_to_cohorts,
-    rank_control_units,
+from diff_diff.staggered import (
+    CallawaySantAnna,
+    CallawaySantAnnaResults,
+    GroupTimeEffect,
 )
 from diff_diff.utils import (
+    WildBootstrapResults,
     check_parallel_trends,
     check_parallel_trends_robust,
     equivalence_test_trends,
-    WildBootstrapResults,
     wild_bootstrap_se,
 )
-from diff_diff.diagnostics import (
-    PlaceboTestResults,
-    run_placebo_test,
-    placebo_timing_test,
-    placebo_group_test,
-    permutation_test,
-    leave_one_out_test,
-    run_all_placebo_tests,
+from diff_diff.visualization import (
+    plot_event_study,
+    plot_group_effects,
+    plot_honest_event_study,
+    plot_sensitivity,
 )
 
-__version__ = "0.5.0"
+__version__ = "0.6.0"
 __all__ = [
     # Estimators
     "DifferenceInDifferences",
@@ -73,6 +85,8 @@ __all__ = [
     # Visualization
     "plot_event_study",
     "plot_group_effects",
+    "plot_sensitivity",
+    "plot_honest_event_study",
     # Parallel trends testing
     "check_parallel_trends",
     "check_parallel_trends_robust",
@@ -99,4 +113,13 @@ __all__ = [
     "create_event_time",
     "aggregate_to_cohorts",
     "rank_control_units",
+    # Honest DiD sensitivity analysis
+    "HonestDiD",
+    "HonestDiDResults",
+    "SensitivityResults",
+    "DeltaSD",
+    "DeltaRM",
+    "DeltaSDRM",
+    "compute_honest_did",
+    "sensitivity_plot",
 ]

diff_diff/diagnostics.py

@@ -360,6 +360,9 @@ def run_placebo_test(
             **estimator_kwargs
         )
 
+    # This should never be reached due to validation above
+    raise ValueError(f"Unknown test type: {test_type}")
+
 
 def placebo_timing_test(
     data: pd.DataFrame,

diff_diff/estimators.py

@@ -603,6 +603,7 @@ class DifferenceInDifferences:
         """
         if not self.is_fitted_:
             raise RuntimeError("Model must be fitted before calling summary()")
+        assert self.results_ is not None
         return self.results_.summary()
 
     def print_summary(self) -> None:
@@ -641,7 +642,7 @@ class TwoWayFixedEffects(DifferenceInDifferences):
     staggered designs.
     """
 
-    def fit(
+    def fit(  # type: ignore[override]
         self,
         data: pd.DataFrame,
         outcome: str,
@@ -763,7 +764,7 @@ class TwoWayFixedEffects(DifferenceInDifferences):
             n_treated=n_treated,
             n_control=n_control,
             alpha=self.alpha,
-            coefficients={"ATT": att},
+            coefficients={"ATT": float(att)},
             vcov=vcov,
             residuals=residuals,
             fitted_values=fitted,
@@ -892,7 +893,7 @@ class MultiPeriodDiD(DifferenceInDifferences):
     The average ATT is computed as the mean of the δ_t coefficients.
     """
 
-    def fit(
+    def fit(  # type: ignore[override]
         self,
         data: pd.DataFrame,
         outcome: str,
@@ -1166,6 +1167,7 @@ class MultiPeriodDiD(DifferenceInDifferences):
         """
         if not self.is_fitted_:
             raise RuntimeError("Model must be fitted before calling summary()")
+        assert self.results_ is not None
         return self.results_.summary()
 
 
@@ -1284,7 +1286,7 @@ class SyntheticDiD(DifferenceInDifferences):
         self._unit_weights = None
         self._time_weights = None
 
-    def fit(
+    def fit(  # type: ignore[override]
         self,
         data: pd.DataFrame,
         outcome: str,