Merge pull request #222 from igerber/enhanced-visualizations

Add enhanced visualizations with plotly backend

Author: igerber

CHANGELOG.md

@@ -7,6 +7,15 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+- Refactor `visualization.py` into `visualization/` subpackage for better organization
+- Add `plot_synth_weights()` — bar chart of Synthetic DiD unit/time weights
+- Add `plot_staircase()` — treatment adoption visualization for staggered designs
+- Add `plot_dose_response()` — dose-response curve for Continuous DiD
+- Add `plot_group_time_heatmap()` — heatmap of ATT(g,t) from staggered estimators
+- Add interactive plotly backend (`backend="plotly"`) for all visualization functions
+- Add `plotly` optional dependency group (`pip install diff-diff[plotly]`)
+
 ## [2.7.4] - 2026-03-21
 
 ### Added

TODO.md

@@ -69,6 +69,7 @@ Deferred items from PR reviews that were not addressed before merge.
 | R comparison tests spawn separate `Rscript` per test (slow CI) | `tests/test_methodology_twfe.py:294` | #139 | Low |
 | CS R helpers hard-code `xformla = ~ 1`; no covariate-adjusted R benchmark for IRLS path | `tests/test_methodology_callaway.py` | #202 | Low |
 | ~376 `duplicate object description` Sphinx warnings — caused by autodoc `:members:` on dataclass attributes within manual API pages (not from autosummary stubs); fix requires restructuring `docs/api/*.rst` pages to avoid documenting the same attribute via both `:members:` and inline `autosummary` tables | `docs/api/*.rst` | — | Low |
+| Plotly renderers silently ignore styling kwargs (marker, markersize, linewidth, capsize, ci_linewidth) that the matplotlib backend honors; thread them through or reject when `backend="plotly"` | `visualization/_event_study.py`, `_diagnostic.py`, `_power.py` | #222 | Medium |
 
 ---
 

diff_diff/__init__.py

@@ -155,12 +155,16 @@
 )
 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,
@@ -220,7 +224,6 @@
     "BaconDecompositionResults",
     "Comparison2x2",
     "bacon_decompose",
-    "plot_bacon",
     # Results
     "DiDResults",
     "MultiPeriodDiDResults",
@@ -251,10 +254,17 @@
     "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 @@
     "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",