diffindiff 2.3.8__tar.gz → 2.4.1__tar.gz

{diffindiff-2.3.8/diffindiff.egg-info → diffindiff-2.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: diffindiff
-Version: 2.3.8
+Version: 2.4.1
 Summary: diffindiff: Python library for convenient Difference-in-Differences analyses
 Author: Thomas Wieland
 Author-email: geowieland@googlemail.com
@@ -8,7 +8,7 @@ Description-Content-Type: text/markdown
 
 # diffindiff: Python library for convenient Difference-in-Differences analyses
 
-This Python library is designed for performing Difference-in-Differences (DiD) analyses in a convenient way. The package is intended to be used in econometric analyses of natural experiments by researchers in economics, marketing, economic geography, and health sciences. It is designed to cover the entire workflow of a DiD analysis, while not requiring extensive programming skills. The package allows users to construct datasets, define treatment and control groups, and set treatment periods. DiD model analyses may be conducted with both datasets created by built-in functions and ready-to-use external datasets. Both simultaneous and staggered adoption are supported. The library allows for various extensions, such as two-way fixed effects models, group- or individual-specific effects, post-treatment periods, and triple-difference estimations. Additionally, it includes functions for visualizing results, such as plotting DiD coefficients with confidence intervals and illustrating the temporal evolution of staggered treatments. Furthermore, several functions for rigorous treatment setting and data diagnostics are incorporated.
+This Python library is designed for performing Difference-in-Differences (DiD) analyses in a convenient way. The package is intended to be used in econometric analyses of natural experiments by researchers in economics, marketing, economic geography, and health sciences. It is designed to cover the entire workflow of a DiD analysis, while not requiring extensive programming skills. The package allows users to construct datasets, define treatment and control groups, and set treatment periods. DiD model analyses may be conducted with both datasets created by built-in functions and ready-to-use external datasets. Both simultaneous and staggered adoption are supported. The library allows for various extensions, such as two-way fixed effects models, group- or individual-specific effects, post-treatment periods, and triple-difference estimations. Additionally, it includes functions for visualizing results, such as plotting DiD coefficients with confidence intervals and illustrating the temporal evolution of staggered treatments. Furthermore, several functions for rigorous treatment configuration and data diagnostics are incorporated.
 
 
 ## Author
@@ -22,12 +22,14 @@ Thomas Wieland [ORCID](https://orcid.org/0000-0001-5168-9846) [EMail](mailto:geo
 - 💻 GitHub Repository: [diffindiff_official](https://github.com/geowieland/diffindiff_official)
 - 📄 DOI (Zenodo): [10.5281/zenodo.18656820](https://doi.org/10.5281/zenodo.18656820)
 
+A research note featuring a case study that utilizes the diffindiff library is available on [arXiv](https://arxiv.org/abs/2605.21464).
+
 
 ## Citation
 
 If you use this software, please cite:
 
-Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.3.8) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
+Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.4.1) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
 
 
 ## Installation
@@ -63,6 +65,7 @@ pip install git+https://github.com/geowieland/diffindiff_official.git
     - Including covariates
     - Including after-treatment period
   - Perform Triple Difference (DDD) analysis
+  - Perform DiD analysis with demeaned data
 - **Diagnosis tools and extensions of analyses**:
   - Add own counterfactuals or create counterfactuals based on machine learning or OLS regression models
   - Bonferroni correction for treatment effects
@@ -166,6 +169,7 @@ See the /tests directory for usage examples of most of the included functions.
   - von Bismarck-Osten C, Borusyak K, Schönberg U (2022) The role of schools in transmission of the SARS-CoV-2 virus: quasi-experimental evidence from Germany. *Economic Policy* 37(109): 87–130. [10.1093/epolic/eiac001](https://doi.org/10.1093/epolic/eiac001)
   - Wieland T (2025) Assessing the effectiveness of non-pharmaceutical interventions in the SARS-CoV-2 pandemic: results of a natural experiment regarding Baden-Württemberg (Germany) and Switzerland in the second infection wave. *Journal of Public Health: From Theory to Practice* 33(11): 2497-2511. [10.1007/s10389-024-02218-x](https://doi.org/10.1007/s10389-024-02218-x)
   - Wooldridge JM (2012) *Introductory Econometrics. A Modern Approach*.
+  - Wooldridge JM (2025) Two-way fixed effects, the two-way mundlak regression, and difference-in-differences estimators. *Empirical Economics* 69(5): 2545-2587. [10.1007/s00181-025-02807-z](https://doi.org/10.1007/s00181-025-02807-z)
 
 
 ## AI Usage Statement
@@ -173,7 +177,7 @@ See the /tests directory for usage examples of most of the included functions.
 This software was developed without the use of AI-generated code. The Continue Agent in Microsoft Visual Studio Code using the GPT-5 mini model (by OpenAI) was used solely to assist in drafting and refining docstrings for documentation. The corresponding guidelines and constraints defined by the author are documented in `AGENTS-docstrings.md` in the [public GitHub repository](https://github.com/geowieland/diffindiff_official).
 
 
-## What's new (v2.3.8)
+## What's new (v2.4.1)
 
-- Bugfixes:
-  - DiffModel.plot() always shows the correct values if retransform_log_outcome is used
+- Other:
+  - Extension of documentation

diffindiff-2.3.8/PKG-INFO → diffindiff-2.4.1/README.md

@@ -1,14 +1,6 @@
-Metadata-Version: 2.1
-Name: diffindiff
-Version: 2.3.8
-Summary: diffindiff: Python library for convenient Difference-in-Differences analyses
-Author: Thomas Wieland
-Author-email: geowieland@googlemail.com
-Description-Content-Type: text/markdown
-
 # diffindiff: Python library for convenient Difference-in-Differences analyses
 
-This Python library is designed for performing Difference-in-Differences (DiD) analyses in a convenient way. The package is intended to be used in econometric analyses of natural experiments by researchers in economics, marketing, economic geography, and health sciences. It is designed to cover the entire workflow of a DiD analysis, while not requiring extensive programming skills. The package allows users to construct datasets, define treatment and control groups, and set treatment periods. DiD model analyses may be conducted with both datasets created by built-in functions and ready-to-use external datasets. Both simultaneous and staggered adoption are supported. The library allows for various extensions, such as two-way fixed effects models, group- or individual-specific effects, post-treatment periods, and triple-difference estimations. Additionally, it includes functions for visualizing results, such as plotting DiD coefficients with confidence intervals and illustrating the temporal evolution of staggered treatments. Furthermore, several functions for rigorous treatment setting and data diagnostics are incorporated.
+This Python library is designed for performing Difference-in-Differences (DiD) analyses in a convenient way. The package is intended to be used in econometric analyses of natural experiments by researchers in economics, marketing, economic geography, and health sciences. It is designed to cover the entire workflow of a DiD analysis, while not requiring extensive programming skills. The package allows users to construct datasets, define treatment and control groups, and set treatment periods. DiD model analyses may be conducted with both datasets created by built-in functions and ready-to-use external datasets. Both simultaneous and staggered adoption are supported. The library allows for various extensions, such as two-way fixed effects models, group- or individual-specific effects, post-treatment periods, and triple-difference estimations. Additionally, it includes functions for visualizing results, such as plotting DiD coefficients with confidence intervals and illustrating the temporal evolution of staggered treatments. Furthermore, several functions for rigorous treatment configuration and data diagnostics are incorporated.
 
 
 ## Author
@@ -22,12 +14,14 @@ Thomas Wieland [ORCID](https://orcid.org/0000-0001-5168-9846) [EMail](mailto:geo
 - 💻 GitHub Repository: [diffindiff_official](https://github.com/geowieland/diffindiff_official)
 - 📄 DOI (Zenodo): [10.5281/zenodo.18656820](https://doi.org/10.5281/zenodo.18656820)
 
+A research note featuring a case study that utilizes the diffindiff library is available on [arXiv](https://arxiv.org/abs/2605.21464).
+
 
 ## Citation
 
 If you use this software, please cite:
 
-Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.3.8) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
+Wieland, T. (2026). diffindiff: A Python library for convenient difference-in-differences analyses (Version 2.4.1) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
 
 
 ## Installation
@@ -63,6 +57,7 @@ pip install git+https://github.com/geowieland/diffindiff_official.git
     - Including covariates
     - Including after-treatment period
   - Perform Triple Difference (DDD) analysis
+  - Perform DiD analysis with demeaned data
 - **Diagnosis tools and extensions of analyses**:
   - Add own counterfactuals or create counterfactuals based on machine learning or OLS regression models
   - Bonferroni correction for treatment effects
@@ -166,6 +161,7 @@ See the /tests directory for usage examples of most of the included functions.
   - von Bismarck-Osten C, Borusyak K, Schönberg U (2022) The role of schools in transmission of the SARS-CoV-2 virus: quasi-experimental evidence from Germany. *Economic Policy* 37(109): 87–130. [10.1093/epolic/eiac001](https://doi.org/10.1093/epolic/eiac001)
   - Wieland T (2025) Assessing the effectiveness of non-pharmaceutical interventions in the SARS-CoV-2 pandemic: results of a natural experiment regarding Baden-Württemberg (Germany) and Switzerland in the second infection wave. *Journal of Public Health: From Theory to Practice* 33(11): 2497-2511. [10.1007/s10389-024-02218-x](https://doi.org/10.1007/s10389-024-02218-x)
   - Wooldridge JM (2012) *Introductory Econometrics. A Modern Approach*.
+  - Wooldridge JM (2025) Two-way fixed effects, the two-way mundlak regression, and difference-in-differences estimators. *Empirical Economics* 69(5): 2545-2587. [10.1007/s00181-025-02807-z](https://doi.org/10.1007/s00181-025-02807-z)
 
 
 ## AI Usage Statement
@@ -173,7 +169,7 @@ See the /tests directory for usage examples of most of the included functions.
 This software was developed without the use of AI-generated code. The Continue Agent in Microsoft Visual Studio Code using the GPT-5 mini model (by OpenAI) was used solely to assist in drafting and refining docstrings for documentation. The corresponding guidelines and constraints defined by the author are documented in `AGENTS-docstrings.md` in the [public GitHub repository](https://github.com/geowieland/diffindiff_official).
 
 
-## What's new (v2.3.8)
+## What's new (v2.4.1)
 
-- Bugfixes:
-  - DiffModel.plot() always shows the correct values if retransform_log_outcome is used
+- Other:
+  - Extension of documentation

{diffindiff-2.3.8 → diffindiff-2.4.1}/diffindiff/config.py

@@ -4,15 +4,15 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     1.0.16
-# Last update: 2026-04-21 20:19
+# Version:     1.0.19
+# Last update: 2026-05-22 07:52
 # Copyright (c) 2025-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 
 # Basic config:
 
 PACKAGE_NAME = "diffindiff"
-PACKAGE_VERSION = "2.3.8"
+PACKAGE_VERSION = "2.4.1"
 
 VERBOSE = False
 
@@ -25,8 +25,13 @@ ACCEPT_CONTINUOUS_TREATMENTS = True
 
 # Description texts:
 
-DID_DESCRIPTION = "Difference-in-Differences Analysis"
-DDD_DESCRIPTION = "Triple-Difference Analysis"
+DID = "Difference-in-Differences"
+DDD = "Triple-Difference"
+
+DID_DESCRIPTION = f"{DID} Analysis"
+DDD_DESCRIPTION = f"{DDD} Analysis"
+PLACEBO_DESCRIPTION = "Placebo"
+PREDICTIVE_MODEL_DESCRIPTION = "Predictive Model"
 
 TREATMENT_DESCRIPTION = "Treatment"
 CONTROL_DESCRIPTION = "Control"
@@ -90,6 +95,7 @@ DUMMY_PREFIX = "DUMMY"
 LOG_PREFIX = "log"
 OBSERVED_SUFFIX = "observed"
 EXPECTED_SUFFIX = "expected"
+DEMEAN_SUFFIX = "demean"
 PREDICTED_SUFFIX = "pred"
 CI_LOWER_SUFFIX = "CI_lower"
 CI_UPPER_SUFFIX = "CI_upper"
@@ -98,54 +104,63 @@ PI_UPPER_SUFFIX = "PI_upper"
 SPILLOVER_PREFIX = "Spillover"
 SPILLOVER_UNIT_PREFIX = f"{SPILLOVER_PREFIX}{DELIMITER}{UNIT_COL}"
 DIFFERENCE_SUFFIX = "diff"
+RESIDUALS_SUFFIX = "resid"
+PLACEBO_PREFIX = "Placebo"
 
 # Modeling config:
 
 # Coefficients/effects types:
 
-TREATMENT_EFFECTS_DESCRIPTION = "Difference-in-Differences coefficients"
+TREATMENT_EFFECTS_DESCRIPTION = f"{DID} coefficients"
 
 EFFECTS_TYPES = {
     "ATE": {
         "description": "Average treatment effect",
         "model_results_key": "average_treatment_effects",
         "summary_treatment_effects": True,
-        "summary_description": "{description} {coef}"
+        "summary_description": "{description} {coef}",
+        "derive_fixed_effects": True
     },
     "AATE": {
         "description": "Average after-treatment effect",
         "model_results_key": "average_after_treatment_effects",
         "summary_treatment_effects": True,
-        "summary_description": "{description} {coef}"
+        "summary_description": "{description} {coef}",
+        "derive_fixed_effects": True
     },    
     "beta_0": {
         "description": "Control group baseline",
         "model_results_key": "control_group_baseline",
         "summary_treatment_effects": True,
-        "summary_description": "{description}"
+        "summary_description": "{description}",
+        "derive_fixed_effects": True
     },
     "beta_1": {
         "description": f"{TREATMENT_GROUP_DESCRIPTION} deviation",
         "model_results_key": "treatment_group_deviation",
         "summary_treatment_effects": True,
-        "summary_description": "{description}"
+        "summary_description": "{description}",
+        "derive_fixed_effects": True
     },
     "delta_0": {
         "description": "Non-treatment time effect",
         "model_results_key": "non_treatment_time_effect",
         "summary_treatment_effects": True,
-        "summary_description": "{description} {coef}"
+        "summary_description": "{description} {coef}",
+        "derive_fixed_effects": True
     },
     "ATT": {
         "description": "After-treatment time effect",
         "model_results_key": "after_treatment_time_effects",
         "summary_treatment_effects": True,
-        "summary_description": "{description} {coef}"
+        "summary_description": "{description} {coef}",
+        "derive_fixed_effects": True
     },
     "FE": {
         "description": "Fixed effects",
         "model_results_key": "fixed_effects",
         "summary_treatment_effects": False,
+        "derive_fixed_effects": False,
         "types": {
             0: {
                 "FE": "unit",
@@ -174,50 +189,57 @@ EFFECTS_TYPES = {
         "description": "Individual time trends",
         "model_results_key": "individual_time_trends",
         "model_config_key": "ITT",
-        "summary_treatment_effects": False
+        "summary_treatment_effects": False,
+        "derive_fixed_effects": False
     },
     "ITE": {
         "description": "Individual treatment effects",
         "model_results_key": "individual_treatment_effects",
         "model_config_key": "ITE",
         "summary_treatment_effects": True,
-        "summary_description": "{coef}"
+        "summary_description": "{coef}",
+        "derive_fixed_effects": False
     },
     "GTT": {
         "description": "Group time trends",
         "model_results_key": "group_time_trends",
         "model_config_key": "GTT",
-        "summary_treatment_effects": False
+        "summary_treatment_effects": False,
+        "derive_fixed_effects": False
     },
     "GTE": {
         "description": "Group treatment effects",
         "model_results_key": "group_treatment_effects",
         "model_config_key": "GTE",
         "summary_treatment_effects": True,
-        "summary_description": "{coef}"
+        "summary_description": "{coef}",
+        "derive_fixed_effects": False
     },
     "spillover": {
         "description": "Treatment spillover effect",
         "model_results_key": "treatment_spillover_effects",
         "model_config_key": "spillover_effects",
         "summary_treatment_effects": True,
-        "summary_description": "{description} {coef}"
+        "summary_description": "{description} {coef}",
+        "derive_fixed_effects": True
     },
     "covariates": {
         "description": "Covariates",
         "model_results_key": "covariates_effects",
         "model_config_key": "covariates",
         "summary_treatment_effects": False,
-        "summary_description": "{coef}"
+        "summary_description": "{coef}",
+        "derive_fixed_effects": True
     }, 
 }
 EFFECTS_TYPES_MODEL_RESULTS = [value["model_results_key"] for value in EFFECTS_TYPES.values() if "model_results_key" in value]
 EFFECTS_TYPES_MODEL_RESULTS_SUMMARY = [value["model_results_key"] for value in EFFECTS_TYPES.values() if "model_results_key" in value and value["summary_treatment_effects"]]
+EFFECTS_TYPES_DERIVE_FE = [value["model_results_key"] for value in EFFECTS_TYPES.values() if "model_results_key" in value and value["derive_fixed_effects"]]
 
 
 EFFECTS_TYPES_DDD = {
     "TDATE": {
-        "description": "Triple-Difference Average treatment effect",
+        "description": f"{DDD} Average treatment effect",
         "model_results_key": "TDATE"
     },
     "beta_2": {
@@ -374,10 +396,13 @@ MODEL_WRAPPER_AVAILABLE = {
     "knn": "K-nearest neighbor",
     "xgb": "Extreme Gradient Boosting",
     "lgbm": "LightGBM",
-    "svr": "Support-vector regression",
+    "svr": "Support-Vector Regression",
+    "mlp": "Multi-layer Perceptron"
 }
 MODEL_WRAPPER_AVAILABLE_LIST = list(MODEL_WRAPPER_AVAILABLE.keys())
 
+MODEL_WRAPPER_DEFAULT = MODEL_WRAPPER_AVAILABLE_LIST[0]
+
 
 # Treatment diagnostics: