Add new analysis modules (bartik, causal_impact, diagnostics, dml, matching, mediation, panel, synth) and modelsummary output

New modules: shift-share/Bartik IV, causal impact analysis, sensitivity diagnostics,
double ML, propensity score matching, causal mediation, panel regression, and
synthetic control. Added modelsummary output formatter and comprehensive tests.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: brycewang-stanford

pyproject.toml

@@ -9,26 +9,32 @@ description = "The AI-powered Statistics & Econometrics Toolkit for Python"
 readme = "README.md"
 license = "MIT"
 authors = [
-    {name = "Bryce Wang", email = "bryce.wang@example.com"}
+    {name = "Bryce Wang", email = "bryce@copaper.ai"}
 ]
 maintainers = [
-    {name = "Bryce Wang", email = "bryce.wang@example.com"}
+    {name = "Bryce Wang", email = "bryce@copaper.ai"}
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",
     "Intended Audience :: Science/Research",
     "Operating System :: OS Independent",
     "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.8",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
     "Topic :: Scientific/Engineering :: Mathematics",
     "Topic :: Office/Business :: Financial",
 ]
-keywords = ["econometrics", "statistics", "regression", "causal-inference", "causal-forest", "panel-data", "instrumental-variables", "stata", "R", "machine-learning"]
-requires-python = ">=3.8"
+keywords = [
+    "econometrics", "statistics", "causal-inference",
+    "difference-in-differences", "regression-discontinuity",
+    "synthetic-control", "matching", "propensity-score",
+    "instrumental-variables", "panel-data", "double-ml",
+    "stata", "R", "machine-learning",
+]
+requires-python = ">=3.9"
 dependencies = [
     "numpy>=1.20.0",
     "pandas>=1.3.0",
@@ -40,6 +46,7 @@ dependencies = [
     "patsy>=0.5.0",
     "openpyxl>=3.0.0",
     "xlsxwriter>=3.0.0",
+    "python-docx>=1.0.0",
 ]
 
 [project.optional-dependencies]
@@ -66,10 +73,10 @@ plotting = [
 ]
 
 [project.urls]
-Homepage = "https://github.com/brycewang-stanford/pyEconometrics"
+Homepage = "https://github.com/brycewang-stanford/statspai"
 Documentation = "https://statspai.readthedocs.io/"
-Repository = "https://github.com/brycewang-stanford/pyEconometrics"
-"Bug Reports" = "https://github.com/brycewang-stanford/pyEconometrics/issues"
+Repository = "https://github.com/brycewang-stanford/statspai"
+"Bug Reports" = "https://github.com/brycewang-stanford/statspai/issues"
 
 [tool.setuptools.packages.find]
 where = ["src"]
@@ -79,7 +86,7 @@ where = ["src"]
 
 [tool.black]
 line-length = 88
-target-version = ['py38']
+target-version = ['py39']
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]
@@ -89,7 +96,7 @@ python_functions = ["test_*"]
 addopts = "--cov=statspai --cov-report=html --cov-report=term-missing"
 
 [tool.mypy]
-python_version = "3.8"
+python_version = "3.9"
 warn_return_any = true
 warn_unused_configs = true
 disallow_untyped_defs = true

src/statspai/__init__.py

@@ -32,7 +32,16 @@
 from .causal.causal_forest import CausalForest, causal_forest
 from .did import did, did_2x2, callaway_santanna
 from .rd import rdrobust, rdplot
+from .synth import synth, SyntheticControl
+from .matching import match, MatchEstimator
+from .dml import dml, DoubleML
+from .panel import panel, PanelRegression
+from .causal_impact import causal_impact, CausalImpactEstimator
+from .mediation import mediate, MediationAnalysis
+from .bartik import bartik, BartikIV
 from .output.outreg2 import OutReg2, outreg2
+from .output.modelsummary import modelsummary, coefplot
+from .diagnostics import oster_bounds, mccrary_test
 
 __all__ = [
     # Core
@@ -49,10 +58,36 @@
     # RD
     "rdrobust",
     "rdplot",
+    # Synthetic Control
+    "synth",
+    "SyntheticControl",
+    # Matching
+    "match",
+    "MatchEstimator",
+    # Double ML
+    "dml",
+    "DoubleML",
+    # Panel
+    "panel",
+    "PanelRegression",
+    # Causal Impact
+    "causal_impact",
+    "CausalImpactEstimator",
     # Causal Forest
     "CausalForest",
     "causal_forest",
     # Output
     "OutReg2",
     "outreg2",
+    "modelsummary",
+    "coefplot",
+    # Mediation
+    "mediate",
+    "MediationAnalysis",
+    # Bartik IV
+    "bartik",
+    "BartikIV",
+    # Diagnostics
+    "oster_bounds",
+    "mccrary_test",
 ]

src/statspai/bartik/__init__.py

@@ -0,0 +1,21 @@
+"""
+Shift-Share (Bartik) Instrumental Variables for StatsPAI.
+
+Constructs Bartik instruments from industry shares and national shocks,
+with diagnostics for instrument validity following Goldsmith-Pinkham,
+Sorkin, and Swift (2020) and Borusyak, Hull, and Jaravel (2022).
+
+References
+----------
+Goldsmith-Pinkham, P., Sorkin, I., and Swift, H. (2020).
+"Bartik Instruments: What, When, Why, and How."
+*American Economic Review*, 110(8), 2586-2624.
+
+Borusyak, K., Hull, P., and Jaravel, X. (2022).
+"Quasi-Experimental Shift-Share Research Designs."
+*Review of Economic Studies*, 89(1), 181-213.
+"""
+
+from .shift_share import bartik, BartikIV
+
+__all__ = ['bartik', 'BartikIV']