diff-diff 2.7.4__tar.gz → 2.7.5__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diff-diff
-Version: 2.7.4
+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.4 → diff_diff-2.7.5}/diff_diff/__init__.py

@@ -155,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,
@@ -187,7 +191,7 @@ Stacked = StackedDiD
 Bacon = BaconDecomposition
 EDiD = EfficientDiD
 
-__version__ = "2.7.4"
+__version__ = "2.7.5"
 __all__ = [
     # Estimators
     "DifferenceInDifferences",
@@ -220,7 +224,6 @@ __all__ = [
     "BaconDecompositionResults",
     "Comparison2x2",
     "bacon_decompose",
-    "plot_bacon",
     # Results
     "DiDResults",
     "MultiPeriodDiDResults",
@@ -251,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",
@@ -308,14 +318,12 @@ __all__ = [
     "simulate_mde",
     "simulate_power",
     "simulate_sample_size",
-    "plot_power_curve",
     # Pre-trends power analysis
     "PreTrendsPower",
     "PreTrendsPowerResults",
     "PreTrendsPowerCurve",
     "compute_pretrends_power",
     "compute_mdv",
-    "plot_pretrends_power",
     # Survey support
     "SurveyDesign",
     "SurveyMetadata",