diff-diff 2.7.3__tar.gz → 2.7.5__tar.gz

{diff_diff-2.7.3 → diff_diff-2.7.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diff-diff
-Version: 2.7.3
+Version: 2.7.5
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Science/Research
 Classifier: Operating System :: OS Independent
@@ -24,14 +24,18 @@ Requires-Dist: ruff>=0.1.0 ; extra == 'dev'
 Requires-Dist: mypy>=1.0 ; extra == 'dev'
 Requires-Dist: maturin>=1.4,<2.0 ; extra == 'dev'
 Requires-Dist: matplotlib>=3.5 ; extra == 'dev'
+Requires-Dist: nbmake>=1.5 ; extra == 'dev'
+Requires-Dist: plotly>=5.0 ; extra == 'dev'
 Requires-Dist: sphinx>=6.0 ; extra == 'docs'
 Requires-Dist: pydata-sphinx-theme>=0.15 ; extra == 'docs'
 Requires-Dist: sphinxext-opengraph>=0.9 ; extra == 'docs'
 Requires-Dist: sphinx-sitemap>=2.5 ; extra == 'docs'
 Requires-Dist: nbsphinx>=0.9 ; extra == 'docs'
 Requires-Dist: matplotlib>=3.5 ; extra == 'docs'
+Requires-Dist: plotly>=5.0 ; extra == 'plotly'
 Provides-Extra: dev
 Provides-Extra: docs
+Provides-Extra: plotly
 Summary: Difference-in-Differences causal inference with sklearn-like API. Callaway-Sant'Anna, Synthetic DiD, Honest DiD, event studies, parallel trends.
 Keywords: causal-inference,difference-in-differences,econometrics,statistics,treatment-effects,event-study,staggered-adoption,parallel-trends,synthetic-control,panel-data,did,twfe,callaway-santanna,honest-did,sensitivity-analysis
 Author: diff-diff contributors

{diff_diff-2.7.3 → diff_diff-2.7.5}/diff_diff/__init__.py

@@ -53,11 +53,15 @@ from diff_diff.honest_did import (
 from diff_diff.power import (
     PowerAnalysis,
     PowerResults,
+    SimulationMDEResults,
     SimulationPowerResults,
+    SimulationSampleSizeResults,
     compute_mde,
     compute_power,
     compute_sample_size,
+    simulate_mde,
     simulate_power,
+    simulate_sample_size,
 )
 from diff_diff.pretrends import (
     PreTrendsPower,
@@ -90,6 +94,10 @@ from diff_diff.results import (
     PeriodEffect,
     SyntheticDiDResults,
 )
+from diff_diff.survey import (
+    SurveyDesign,
+    SurveyMetadata,
+)
 from diff_diff.staggered import (
     CallawaySantAnna,
     CallawaySantAnnaResults,
@@ -147,12 +155,16 @@ from diff_diff.utils import (
 )
 from diff_diff.visualization import (
     plot_bacon,
+    plot_dose_response,
     plot_event_study,
     plot_group_effects,
+    plot_group_time_heatmap,
     plot_honest_event_study,
     plot_power_curve,
     plot_pretrends_power,
     plot_sensitivity,
+    plot_staircase,
+    plot_synth_weights,
 )
 from diff_diff.datasets import (
     clear_cache,
@@ -179,7 +191,7 @@ Stacked = StackedDiD
 Bacon = BaconDecomposition
 EDiD = EfficientDiD
 
-__version__ = "2.7.3"
+__version__ = "2.7.5"
 __all__ = [
     # Estimators
     "DifferenceInDifferences",
@@ -212,7 +224,6 @@ __all__ = [
     "BaconDecompositionResults",
     "Comparison2x2",
     "bacon_decompose",
-    "plot_bacon",
     # Results
     "DiDResults",
     "MultiPeriodDiDResults",
@@ -243,10 +254,17 @@ __all__ = [
     "EDiDBootstrapResults",
     "EDiD",
     # Visualization
+    "plot_bacon",
     "plot_event_study",
     "plot_group_effects",
     "plot_sensitivity",
     "plot_honest_event_study",
+    "plot_power_curve",
+    "plot_pretrends_power",
+    "plot_synth_weights",
+    "plot_staircase",
+    "plot_dose_response",
+    "plot_group_time_heatmap",
     # Parallel trends testing
     "check_parallel_trends",
     "check_parallel_trends_robust",
@@ -291,19 +309,24 @@ __all__ = [
     # Power analysis
     "PowerAnalysis",
     "PowerResults",
+    "SimulationMDEResults",
     "SimulationPowerResults",
+    "SimulationSampleSizeResults",
     "compute_mde",
     "compute_power",
     "compute_sample_size",
+    "simulate_mde",
     "simulate_power",
-    "plot_power_curve",
+    "simulate_sample_size",
     # Pre-trends power analysis
     "PreTrendsPower",
     "PreTrendsPowerResults",
     "PreTrendsPowerCurve",
     "compute_pretrends_power",
     "compute_mdv",
-    "plot_pretrends_power",
+    # Survey support
+    "SurveyDesign",
+    "SurveyMetadata",
     # Rust backend
     "HAS_RUST_BACKEND",
     # Linear algebra helpers