diff-diff 2.1.0__cp39-cp39-macosx_11_0_arm64.whl

diff_diff/__init__.py

@@ -0,0 +1,234 @@
+"""
+diff-diff: A library for Difference-in-Differences analysis.
+
+This library provides sklearn-like estimators for causal inference
+using the difference-in-differences methodology.
+"""
+
+# Import backend detection from dedicated module (avoids circular imports)
+from diff_diff._backend import (
+    HAS_RUST_BACKEND,
+    _rust_bootstrap_weights,
+    _rust_compute_robust_vcov,
+    _rust_project_simplex,
+    _rust_solve_ols,
+    _rust_synthetic_weights,
+)
+
+from diff_diff.bacon import (
+    BaconDecomposition,
+    BaconDecompositionResults,
+    Comparison2x2,
+    bacon_decompose,
+)
+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.linalg import (
+    InferenceResult,
+    LinearRegression,
+)
+from diff_diff.estimators import (
+    DifferenceInDifferences,
+    MultiPeriodDiD,
+    SyntheticDiD,
+    TwoWayFixedEffects,
+)
+from diff_diff.honest_did import (
+    DeltaRM,
+    DeltaSD,
+    DeltaSDRM,
+    HonestDiD,
+    HonestDiDResults,
+    SensitivityResults,
+    compute_honest_did,
+    sensitivity_plot,
+)
+from diff_diff.power import (
+    PowerAnalysis,
+    PowerResults,
+    SimulationPowerResults,
+    compute_mde,
+    compute_power,
+    compute_sample_size,
+    simulate_power,
+)
+from diff_diff.pretrends import (
+    PreTrendsPower,
+    PreTrendsPowerCurve,
+    PreTrendsPowerResults,
+    compute_mdv,
+    compute_pretrends_power,
+)
+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,
+    MultiPeriodDiDResults,
+    PeriodEffect,
+    SyntheticDiDResults,
+)
+from diff_diff.staggered import (
+    CallawaySantAnna,
+    CallawaySantAnnaResults,
+    CSBootstrapResults,
+    GroupTimeEffect,
+)
+from diff_diff.sun_abraham import (
+    SABootstrapResults,
+    SunAbraham,
+    SunAbrahamResults,
+)
+from diff_diff.triple_diff import (
+    TripleDifference,
+    TripleDifferenceResults,
+    triple_difference,
+)
+from diff_diff.trop import (
+    TROP,
+    TROPResults,
+    trop,
+)
+from diff_diff.utils import (
+    WildBootstrapResults,
+    check_parallel_trends,
+    check_parallel_trends_robust,
+    equivalence_test_trends,
+    wild_bootstrap_se,
+)
+from diff_diff.visualization import (
+    plot_bacon,
+    plot_event_study,
+    plot_group_effects,
+    plot_honest_event_study,
+    plot_power_curve,
+    plot_pretrends_power,
+    plot_sensitivity,
+)
+from diff_diff.datasets import (
+    clear_cache,
+    list_datasets,
+    load_card_krueger,
+    load_castle_doctrine,
+    load_dataset,
+    load_divorce_laws,
+    load_mpdta,
+)
+
+__version__ = "2.1.0"
+__all__ = [
+    # Estimators
+    "DifferenceInDifferences",
+    "TwoWayFixedEffects",
+    "MultiPeriodDiD",
+    "SyntheticDiD",
+    "CallawaySantAnna",
+    "SunAbraham",
+    "TripleDifference",
+    "TROP",
+    # Bacon Decomposition
+    "BaconDecomposition",
+    "BaconDecompositionResults",
+    "Comparison2x2",
+    "bacon_decompose",
+    "plot_bacon",
+    # Results
+    "DiDResults",
+    "MultiPeriodDiDResults",
+    "SyntheticDiDResults",
+    "PeriodEffect",
+    "CallawaySantAnnaResults",
+    "CSBootstrapResults",
+    "GroupTimeEffect",
+    "SunAbrahamResults",
+    "SABootstrapResults",
+    "TripleDifferenceResults",
+    "triple_difference",
+    "TROPResults",
+    "trop",
+    # Visualization
+    "plot_event_study",
+    "plot_group_effects",
+    "plot_sensitivity",
+    "plot_honest_event_study",
+    # Parallel trends testing
+    "check_parallel_trends",
+    "check_parallel_trends_robust",
+    "equivalence_test_trends",
+    # Wild cluster bootstrap
+    "WildBootstrapResults",
+    "wild_bootstrap_se",
+    # Placebo tests / diagnostics
+    "PlaceboTestResults",
+    "run_placebo_test",
+    "placebo_timing_test",
+    "placebo_group_test",
+    "permutation_test",
+    "leave_one_out_test",
+    "run_all_placebo_tests",
+    # Data preparation utilities
+    "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",
+    # Honest DiD sensitivity analysis
+    "HonestDiD",
+    "HonestDiDResults",
+    "SensitivityResults",
+    "DeltaSD",
+    "DeltaRM",
+    "DeltaSDRM",
+    "compute_honest_did",
+    "sensitivity_plot",
+    # Power analysis
+    "PowerAnalysis",
+    "PowerResults",
+    "SimulationPowerResults",
+    "compute_mde",
+    "compute_power",
+    "compute_sample_size",
+    "simulate_power",
+    "plot_power_curve",
+    # Pre-trends power analysis
+    "PreTrendsPower",
+    "PreTrendsPowerResults",
+    "PreTrendsPowerCurve",
+    "compute_pretrends_power",
+    "compute_mdv",
+    "plot_pretrends_power",
+    # Rust backend
+    "HAS_RUST_BACKEND",
+    # Linear algebra helpers
+    "LinearRegression",
+    "InferenceResult",
+    # Datasets
+    "load_card_krueger",
+    "load_castle_doctrine",
+    "load_divorce_laws",
+    "load_mpdta",
+    "load_dataset",
+    "list_datasets",
+    "clear_cache",
+]

diff_diff/_backend.py

@@ -0,0 +1,64 @@
+"""
+Backend detection and configuration for diff-diff.
+
+This module handles:
+1. Detection of optional Rust backend
+2. Environment variable configuration (DIFF_DIFF_BACKEND)
+3. Exports HAS_RUST_BACKEND and Rust function references
+
+Other modules should import from here to avoid circular imports with __init__.py.
+"""
+
+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()
+
+# Try to import Rust backend for accelerated operations
+try:
+    from diff_diff._rust_backend import (
+        generate_bootstrap_weights_batch as _rust_bootstrap_weights,
+        compute_synthetic_weights as _rust_synthetic_weights,
+        project_simplex as _rust_project_simplex,
+        solve_ols as _rust_solve_ols,
+        compute_robust_vcov as _rust_compute_robust_vcov,
+    )
+    _rust_available = True
+except ImportError:
+    _rust_available = False
+    _rust_bootstrap_weights = None
+    _rust_synthetic_weights = None
+    _rust_project_simplex = None
+    _rust_solve_ols = None
+    _rust_compute_robust_vcov = None
+
+# Determine final backend based on environment variable and availability
+if _backend_env == 'python':
+    # Force pure Python mode - disable Rust even if available
+    HAS_RUST_BACKEND = False
+    _rust_bootstrap_weights = None
+    _rust_synthetic_weights = None
+    _rust_project_simplex = None
+    _rust_solve_ols = None
+    _rust_compute_robust_vcov = None
+elif _backend_env == 'rust':
+    # Force Rust mode - fail if not available
+    if not _rust_available:
+        raise ImportError(
+            "DIFF_DIFF_BACKEND=rust but Rust backend is not available. "
+            "Install with: pip install diff-diff[rust]"
+        )
+    HAS_RUST_BACKEND = True
+else:
+    # Auto mode - use Rust if available
+    HAS_RUST_BACKEND = _rust_available
+
+__all__ = [
+    'HAS_RUST_BACKEND',
+    '_rust_bootstrap_weights',
+    '_rust_synthetic_weights',
+    '_rust_project_simplex',
+    '_rust_solve_ols',
+    '_rust_compute_robust_vcov',
+]