diff-diff 2.7.4__tar.gz → 2.7.6__tar.gz

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

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

@@ -95,8 +95,10 @@ from diff_diff.results import (
     SyntheticDiDResults,
 )
 from diff_diff.survey import (
+    DEFFDiagnostics,
     SurveyDesign,
     SurveyMetadata,
+    compute_deff_diagnostics,
 )
 from diff_diff.staggered import (
     CallawaySantAnna,
@@ -155,13 +157,18 @@ 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.practitioner import practitioner_next_steps
 from diff_diff.datasets import (
     clear_cache,
     list_datasets,
@@ -187,7 +194,7 @@ Stacked = StackedDiD
 Bacon = BaconDecomposition
 EDiD = EfficientDiD
 
-__version__ = "2.7.4"
+__version__ = "2.7.6"
 __all__ = [
     # Estimators
     "DifferenceInDifferences",
@@ -220,7 +227,6 @@ __all__ = [
     "BaconDecompositionResults",
     "Comparison2x2",
     "bacon_decompose",
-    "plot_bacon",
     # Results
     "DiDResults",
     "MultiPeriodDiDResults",
@@ -251,10 +257,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,17 +321,17 @@ __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",
+    "DEFFDiagnostics",
+    "compute_deff_diagnostics",
     # Rust backend
     "HAS_RUST_BACKEND",
     # Linear algebra helpers
@@ -332,4 +345,6 @@ __all__ = [
     "load_dataset",
     "list_datasets",
     "clear_cache",
+    # Practitioner guidance
+    "practitioner_next_steps",
 ]

{diff_diff-2.7.4 → diff_diff-2.7.6}/diff_diff/_backend.py

@@ -13,7 +13,7 @@ import os
 
 # Check for backend override via environment variable
 # DIFF_DIFF_BACKEND can be: 'auto' (default), 'python', or 'rust'
-_backend_env = os.environ.get('DIFF_DIFF_BACKEND', 'auto').lower()
+_backend_env = os.environ.get("DIFF_DIFF_BACKEND", "auto").lower()
 
 # Try to import Rust backend for accelerated operations
 try:
@@ -38,6 +38,7 @@ try:
         # Diagnostics
         rust_backend_info as _rust_backend_info,
     )
+
     _rust_available = True
 except ImportError:
     _rust_available = False
@@ -61,7 +62,7 @@ except ImportError:
     _rust_backend_info = None
 
 # Determine final backend based on environment variable and availability
-if _backend_env == 'python':
+if _backend_env == "python":
     # Force pure Python mode - disable Rust even if available
     HAS_RUST_BACKEND = False
     _rust_bootstrap_weights = None
@@ -82,7 +83,7 @@ if _backend_env == 'python':
     _rust_compute_noise_level = None
     _rust_sc_weight_fw = None
     _rust_backend_info = None
-elif _backend_env == 'rust':
+elif _backend_env == "rust":
     # Force Rust mode - fail if not available
     if not _rust_available:
         raise ImportError(
@@ -111,23 +112,23 @@ def rust_backend_info():
 
 
 __all__ = [
-    'HAS_RUST_BACKEND',
-    'rust_backend_info',
-    '_rust_bootstrap_weights',
-    '_rust_synthetic_weights',
-    '_rust_project_simplex',
-    '_rust_solve_ols',
-    '_rust_compute_robust_vcov',
+    "HAS_RUST_BACKEND",
+    "rust_backend_info",
+    "_rust_bootstrap_weights",
+    "_rust_synthetic_weights",
+    "_rust_project_simplex",
+    "_rust_solve_ols",
+    "_rust_compute_robust_vcov",
     # TROP estimator acceleration (local method)
-    '_rust_unit_distance_matrix',
-    '_rust_loocv_grid_search',
-    '_rust_bootstrap_trop_variance',
+    "_rust_unit_distance_matrix",
+    "_rust_loocv_grid_search",
+    "_rust_bootstrap_trop_variance",
     # TROP estimator acceleration (global method)
-    '_rust_loocv_grid_search_global',
-    '_rust_bootstrap_trop_variance_global',
+    "_rust_loocv_grid_search_global",
+    "_rust_bootstrap_trop_variance_global",
     # SDID weights (Frank-Wolfe matching R's synthdid)
-    '_rust_sdid_unit_weights',
-    '_rust_compute_time_weights',
-    '_rust_compute_noise_level',
-    '_rust_sc_weight_fw',
+    "_rust_sdid_unit_weights",
+    "_rust_compute_time_weights",
+    "_rust_compute_noise_level",
+    "_rust_sc_weight_fw",
 ]