diffindiff 2.1.1__tar.gz → 2.2.1__tar.gz

{diffindiff-2.1.1 → diffindiff-2.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: diffindiff
-Version: 2.1.1
+Version: 2.2.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: Difference-in-Differences (DiD) Analysis Python Library
 
-This Python library is designed for performing Difference-in-Differences (DiD) analyses in a convenient way. It 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, and post-treatment periods. Additionally, it includes functions for visualizing results, such as plotting DiD coefficients with confidence intervals and illustrating the temporal evolution of staggered treatments.
+This Python library is designed for performing Difference-in-Differences (DiD) analyses in a convenient way. It 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.
 
 
 ## Author
@@ -16,11 +16,9 @@ This Python library is designed for performing Difference-in-Differences (DiD) a
 Thomas Wieland [ORCID](https://orcid.org/0000-0001-5168-9846) [EMail](mailto:geowieland@googlemail.com) 
 
 
-## Updates v2.1.1
+## Updates v2.2.1
 - Bugfixes:
-  - Checking input parameters in didanalysis.DiffModel.plot()
-  - Default verbose = False
-  - Notes and warnings related to treatment diagnostics independent of verbose parameter
+  - Always converting unit and time col to str and calculating unit-time-index col
 
 
 ## Features
@@ -62,15 +60,17 @@ Thomas Wieland [ORCID](https://orcid.org/0000-0001-5168-9846) [EMail](mailto:geo
   - Card D, Krueger AD (1994) Minimum Wages and Employment: A Case Study of the Fast Food Industry in New Jersey and Pennsylvania. *The American Economic Review* 84(4): 772-793. [JSTOR](https://www.jstor.org/stable/2677856)
   - de Haas S, Götz G, Heim S (2022) Measuring the effect of COVID‑19‑related night curfews in a bundled intervention within Germany. *Scientific Reports* 12: 19732. [10.1038/s41598-022-24086-9](https://doi.org/10.1038/s41598-022-24086-9)
   - Goodman-Bacon A (2021) Difference-in-differences with variation in treatment timing. *Journal of Econometrics* 225(2): 254-277. [10.1016/j.jeconom.2021.03.014](https://doi.org/10.1016/j.jeconom.2021.03.014)
-  - Greene WH (2012) *Econometric Analysis*. Chapter 6.2.5.
+  - Greene WH (2012) *Econometric Analysis*.
   - Goldfarb A, Tucker C, Wang Y (2022) Conducting Research in Marketing with Quasi-Experiments. *Journal of Marketing* 86(3): 1-19. [10.1177/00222429221082977](https://doi.org/10.1177/00222429221082977)
   - Isporhing IE, Lipfert M, Pestel N (2021) Does re-opening schools contribute to the spread of SARS-CoV-2? Evidence from staggered summer breaks in Germany. *Journal of Public Economics* 198: 104426. [10.1016/j.jpubeco.2021.104426](https://doi.org/10.1016/j.jpubeco.2021.104426)
   - Li KT, Luo L, Pattabhiramaiah A (2024) Causal Inference with Quasi-Experimental Data. *IMPACT at JMR* November 13, 2024. [AMA](https://www.ama.org/marketing-news/causal-inference-with-quasi-experimental-data/)
+  - Olden A (2018) What do you buy when no one's watching? The effect of self-service checkouts on the composition of sales in retail. Discussion paper FOR 3/18, Norwegian School of Economics, Norway. [http://hdl.handle.net/11250/2490886](http://hdl.handle.net/11250/2490886)
   - Olden A, Moen J (2022) The triple difference estimator. *The Econometrics Journal* 25(3): 531-553. [10.1093/ectj/utac010](https://doi.org/10.1093/ectj/utac010)
+  - Strassmann A, Çolak Y, Serra-Burriel M, Nordestgaard BG, Turk A, Afzal S, Puhan MA (2023) Nationwide indoor smoking ban and impact on smoking behaviour and lung function: a two-population natural experiment. *Thorax* 78(2): 144-150. [10.1136/thoraxjnl-2021-218436](https://doi.org/10.1136/thoraxjnl-2021-218436)
   - Villa JM (2016) diff: Simplifying the estimation of difference-in-differences treatment effects. *The Stata Journal* 16(1): 52-71. [10.1177/1536867X1601600108](https://doi.org/10.1177/1536867X1601600108)
   - 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 (2024) 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*. [10.1007/s10389-024-02218-x](https://doi.org/10.1007/s10389-024-02218-x)
-  - Wooldridge JM (2012) *Introductory Econometrics. A Modern Approach*. Chapter 13.2.
+  - 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*.
 
 
 ## Examples

{diffindiff-2.1.1 → diffindiff-2.2.1}/README.md

@@ -1,6 +1,6 @@
 # diffindiff: Difference-in-Differences (DiD) Analysis Python Library
 
-This Python library is designed for performing Difference-in-Differences (DiD) analyses in a convenient way. It 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, and post-treatment periods. Additionally, it includes functions for visualizing results, such as plotting DiD coefficients with confidence intervals and illustrating the temporal evolution of staggered treatments.
+This Python library is designed for performing Difference-in-Differences (DiD) analyses in a convenient way. It 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.
 
 
 ## Author
@@ -8,11 +8,9 @@ This Python library is designed for performing Difference-in-Differences (DiD) a
 Thomas Wieland [ORCID](https://orcid.org/0000-0001-5168-9846) [EMail](mailto:geowieland@googlemail.com) 
 
 
-## Updates v2.1.1
+## Updates v2.2.1
 - Bugfixes:
-  - Checking input parameters in didanalysis.DiffModel.plot()
-  - Default verbose = False
-  - Notes and warnings related to treatment diagnostics independent of verbose parameter
+  - Always converting unit and time col to str and calculating unit-time-index col
 
 
 ## Features
@@ -54,15 +52,17 @@ Thomas Wieland [ORCID](https://orcid.org/0000-0001-5168-9846) [EMail](mailto:geo
   - Card D, Krueger AD (1994) Minimum Wages and Employment: A Case Study of the Fast Food Industry in New Jersey and Pennsylvania. *The American Economic Review* 84(4): 772-793. [JSTOR](https://www.jstor.org/stable/2677856)
   - de Haas S, Götz G, Heim S (2022) Measuring the effect of COVID‑19‑related night curfews in a bundled intervention within Germany. *Scientific Reports* 12: 19732. [10.1038/s41598-022-24086-9](https://doi.org/10.1038/s41598-022-24086-9)
   - Goodman-Bacon A (2021) Difference-in-differences with variation in treatment timing. *Journal of Econometrics* 225(2): 254-277. [10.1016/j.jeconom.2021.03.014](https://doi.org/10.1016/j.jeconom.2021.03.014)
-  - Greene WH (2012) *Econometric Analysis*. Chapter 6.2.5.
+  - Greene WH (2012) *Econometric Analysis*.
   - Goldfarb A, Tucker C, Wang Y (2022) Conducting Research in Marketing with Quasi-Experiments. *Journal of Marketing* 86(3): 1-19. [10.1177/00222429221082977](https://doi.org/10.1177/00222429221082977)
   - Isporhing IE, Lipfert M, Pestel N (2021) Does re-opening schools contribute to the spread of SARS-CoV-2? Evidence from staggered summer breaks in Germany. *Journal of Public Economics* 198: 104426. [10.1016/j.jpubeco.2021.104426](https://doi.org/10.1016/j.jpubeco.2021.104426)
   - Li KT, Luo L, Pattabhiramaiah A (2024) Causal Inference with Quasi-Experimental Data. *IMPACT at JMR* November 13, 2024. [AMA](https://www.ama.org/marketing-news/causal-inference-with-quasi-experimental-data/)
+  - Olden A (2018) What do you buy when no one's watching? The effect of self-service checkouts on the composition of sales in retail. Discussion paper FOR 3/18, Norwegian School of Economics, Norway. [http://hdl.handle.net/11250/2490886](http://hdl.handle.net/11250/2490886)
   - Olden A, Moen J (2022) The triple difference estimator. *The Econometrics Journal* 25(3): 531-553. [10.1093/ectj/utac010](https://doi.org/10.1093/ectj/utac010)
+  - Strassmann A, Çolak Y, Serra-Burriel M, Nordestgaard BG, Turk A, Afzal S, Puhan MA (2023) Nationwide indoor smoking ban and impact on smoking behaviour and lung function: a two-population natural experiment. *Thorax* 78(2): 144-150. [10.1136/thoraxjnl-2021-218436](https://doi.org/10.1136/thoraxjnl-2021-218436)
   - Villa JM (2016) diff: Simplifying the estimation of difference-in-differences treatment effects. *The Stata Journal* 16(1): 52-71. [10.1177/1536867X1601600108](https://doi.org/10.1177/1536867X1601600108)
   - 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 (2024) 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*. [10.1007/s10389-024-02218-x](https://doi.org/10.1007/s10389-024-02218-x)
-  - Wooldridge JM (2012) *Introductory Econometrics. A Modern Approach*. Chapter 13.2.
+  - 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*.
 
 
 ## Examples

diffindiff-2.2.1/diffindiff/config.py

@@ -0,0 +1,478 @@
+#-----------------------------------------------------------------------
+# Name:        config (diffindiff package)
+# Purpose:     Configuration for the diffindiff package
+# Author:      Thomas Wieland 
+#              ORCID: 0000-0001-5168-9846
+#              mail: geowieland@googlemail.com              
+# Version:     1.0.3
+# Last update: 2025-12-06 10:48
+# Copyright (c) 2025 Thomas Wieland
+#-----------------------------------------------------------------------
+
+# Basic config:
+
+PACKAGE_VERSION = "2.2.1"
+
+VERBOSE = True
+
+ROUND_STATISTIC = 3
+ROUND_PERCENT = 2
+
+AUTO_SWITCH_TO_PREPOST = True
+
+# Description texts:
+
+DID_DESCRIPTION = "Difference-in-Differences Analysis"
+DDD_DESCRIPTION = "Triple-Difference Analysis"
+
+TREATMENT_DESCRIPTION = "Treatment"
+
+GROUP_DESCRIPTION = "Group"
+
+TREATMENT_GROUP_DESCRIPTION = f"{TREATMENT_DESCRIPTION} {GROUP_DESCRIPTION}"
+CONTROL_GROUP_DESCRIPTION = f"Control {GROUP_DESCRIPTION}"
+GROUPS_DESCRIPTION = f"{TREATMENT_DESCRIPTION} and {CONTROL_GROUP_DESCRIPTION}"
+
+TIME_PERIODS_DESCRIPTION = "Time periods"
+TREATMENT_PERIOD_DESCRIPTION = f"{TREATMENT_DESCRIPTION} period"
+STUDY_PERIOD_DESCRIPTION = "Study period"
+PREPOST_DESCRIPTION = "Pre-post"
+AFTER_TREATMENT_PERIOD_DESCRIPTION = "After-treatment period"
+
+UNITS_DESCRIPTION = "Units"
+
+N_DESCRIPTION = "Number of observations"
+
+DDD_GROUP_DESCRIPTION = f"{GROUP_DESCRIPTION} segmentation"
+
+TREATMENT_SIMULTANEOUS_DESCRIPTION = "Simultaneous"
+TREATMENT_STAGGERED_DESCRIPTION = "Staggered"
+NO_TREATMENT_CG_DESCRIPTION = f"No-treatment {CONTROL_GROUP_DESCRIPTION}"
+
+PREPOST_PANELDATA_DESCRIPTION = f"{PREPOST_DESCRIPTION} data"
+MULTIPERIOD_PANELDATA_DESCRIPTION = "Multi-period panel data"
+
+DIDDATA_SUMMARY_LABELS = [
+    TREATMENT_DESCRIPTION,
+    UNITS_DESCRIPTION,
+    TREATMENT_GROUP_DESCRIPTION,
+    CONTROL_GROUP_DESCRIPTION,
+    DDD_GROUP_DESCRIPTION,
+    STUDY_PERIOD_DESCRIPTION,
+    TREATMENT_PERIOD_DESCRIPTION,
+    N_DESCRIPTION
+]
+DIDDATA_SUMMARY_MAX_WIDTH = max(len(label) for label in DIDDATA_SUMMARY_LABELS) + 1
+
+# Data management:
+
+DELIMITER = "_"
+DELIMITER_INTERACT = "x"
+
+COL_ABBREV = "col"
+
+DUMMY_PREFIX = "DUMMY"
+LOG_PREFIX = "log"
+OBSERVED_SUFFIX = "observed"
+EXPECTED_SUFFIX = "expected"
+PREDICTED_SUFFIX = "pred"
+CI_LOWER_SUFFIX = "CI_lower"
+CI_UPPER_SUFFIX = "CI_upper"
+PI_LOWER_SUFFIX = "PI_lower"
+PI_UPPER_SUFFIX = "PI_upper"
+
+TG_COL = "TG"
+CG_COL = "CG"
+BG_COL = "BG"
+TT_COL = "TT"
+ATT_COL = "ATT"
+TIME_COL = "t"
+UNIT_COL = "unit"
+UNIT_TIME_COL = f"{UNIT_COL}{DELIMITER}{TIME_COL}"
+TIME_COUNTER_COL = "time_counter"
+TREATMENT_COL = f"{TG_COL}{DELIMITER_INTERACT}{TT_COL}"
+
+
+# Modeling config:
+
+# Coefficients/effects types:
+
+TREATMENT_EFFECTS_DESCRIPTION = "Difference-in-Differences coefficients"
+
+EFFECTS_TYPES = {
+    "ATE": {
+        "description": "Average treatment effect",
+        "model_results_key": "average_treatment_effects",
+        "summary_treatment_effects": True,
+        "summary_description": "{description} {coef}"
+    },
+    "AATE": {
+        "description": "Average after-treatment effect",
+        "model_results_key": "average_after_treatment_effects",
+        "summary_treatment_effects": True,
+        "summary_description": "{description} {coef}"
+    },    
+    "beta_0": {
+        "description": "Control group baseline",
+        "model_results_key": "control_group_baseline",
+        "summary_treatment_effects": True,
+        "summary_description": "{description}"
+    },
+    "beta_1": {
+        "description": f"{TREATMENT_GROUP_DESCRIPTION} deviation",
+        "model_results_key": "treatment_group_deviation",
+        "summary_treatment_effects": True,
+        "summary_description": "{description}"
+    },
+    "delta_0": {
+        "description": "Non-treatment time effect",
+        "model_results_key": "non_treatment_time_effect",
+        "summary_treatment_effects": True,
+        "summary_description": "{description} {coef}"
+    },
+    "ATT": {
+        "description": "After-treatment time effect",
+        "model_results_key": "after_treatment_time_effects",
+        "summary_treatment_effects": True,
+        "summary_description": "{description} {coef}"
+    },
+    "FE": {
+        "description": "Fixed effects",
+        "model_results_key": "fixed_effects",
+        "summary_treatment_effects": False,
+        "types": {
+            0: {
+                "FE": "unit",
+                "dummy_prefix": "UNIT",
+                "model_config_key": "FE_unit",
+                "model_results_key": "FE_unit",
+                "description": "Fixed effects for observational units"
+            },
+            1: {
+                "FE": "time",
+                "dummy_prefix": "TIME",
+                "model_config_key": "FE_time",
+                "model_results_key": "FE_time",
+                "description": "Fixed effects for time points"
+            },
+            2: {
+                "FE": "group",
+                "dummy_prefix": "GROUP",
+                "model_config_key": "FE_group",
+                "model_results_key": "FE_group",
+                "description": "Fixed effects for groups"
+            },
+        }
+    },
+    "ITT": {
+        "description": "Individual time trends",
+        "model_results_key": "individual_time_trends",
+        "model_config_key": "ITT",
+        "summary_treatment_effects": False
+    },
+    "ITE": {
+        "description": "Individual treatment effects",
+        "model_results_key": "individual_treatment_effects",
+        "model_config_key": "ITE",
+        "summary_treatment_effects": True,
+        "summary_description": "{coef}"
+    },
+    "GTT": {
+        "description": "Group time trends",
+        "model_results_key": "group_time_trends",
+        "model_config_key": "GTT",
+        "summary_treatment_effects": False
+    },
+    "GTE": {
+        "description": "Group treatment effects",
+        "model_results_key": "group_treatment_effects",
+        "model_config_key": "GTE",
+        "summary_treatment_effects": True,
+        "summary_description": "{coef}"
+    },
+    "spillover": {
+        "description": "Treatment spillover effect",
+        "model_results_key": "treatment_spillover_effects",
+        "model_config_key": "spillover_effects",
+        "summary_treatment_effects": True,
+        "summary_description": "{description} {coef}"
+    },
+    "covariates": {
+        "description": "Covariates",
+        "model_results_key": "covariates_effects",
+        "model_config_key": "covariates",
+        "summary_treatment_effects": False,
+        "summary_description": "{coef}"
+    }, 
+}
+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_DDD = {
+    "TDATE": {
+        "description": "Triple-Difference Average treatment effect",
+        "model_results_key": "TDATE"
+    },
+    "beta_2": {
+        "description": "Benefit group deviation",
+        "model_results_key": "benefit_group_deviation"
+    },
+    "beta_4": {
+        "description": "Treated benefit group deviation",
+        "model_results_key": "treated_benefit_group_deviation"
+    },
+    "beta_6": {
+        "description": "Benefit group non-treatment time effect",
+        "model_results_key": "benefit_non_treatment_time_effect"
+    },
+    list(EFFECTS_TYPES.keys())[0]: EFFECTS_TYPES[list(EFFECTS_TYPES.keys())[0]],
+    list(EFFECTS_TYPES.keys())[2]: EFFECTS_TYPES[list(EFFECTS_TYPES.keys())[2]],
+    list(EFFECTS_TYPES.keys())[3]: EFFECTS_TYPES[list(EFFECTS_TYPES.keys())[3]],
+    list(EFFECTS_TYPES.keys())[4]: EFFECTS_TYPES[list(EFFECTS_TYPES.keys())[4]],
+    list(EFFECTS_TYPES.keys())[6]: EFFECTS_TYPES[list(EFFECTS_TYPES.keys())[6]],
+}
+
+EFFECTS_TYPES_DDD_MODEL_RESULTS = [value["model_results_key"] for value in EFFECTS_TYPES_DDD.values() if "model_results_key" in value]
+
+FE_TYPES = [value["FE"] for value in EFFECTS_TYPES["FE"]["types"].values()]
+
+# Time trends:
+TIME_TRENDS_TYPES = [
+    list(EFFECTS_TYPES.keys())[7],
+    list(EFFECTS_TYPES.keys())[9]
+    ]
+
+# Specific effects:
+SPECIFIC_EFFFECTS_TYPES = [
+    list(EFFECTS_TYPES.keys())[8],
+    list(EFFECTS_TYPES.keys())[10]
+    ]
+
+OLS_MODEL_RESULTS = {
+    "coef_name": {
+        "model_results_key": "Coefficient",
+        "summary_description": "Coefficient",
+    },
+    "coef": {
+        "model_results_key": "Estimate",
+        "summary_description": "Estimate",
+    },
+    "coef_standard_errors": {
+        "model_results_key": "SE",
+        "summary_description": "SE",
+    },
+    "coef_teststatistic": {
+        "model_results_key": "t",
+        "summary_description": "t"
+    },
+    "coef_p": {
+        "model_results_key": "p",
+        "summary_description": "p"
+    },
+    "coef_confint_lower": {
+        "model_results_key": "CI_lower",
+        "summary_description": "CI lower"
+    },
+    "coef_confint_upper": {
+        "model_results_key": "CI_upper",
+        "summary_description": "CI upper"
+    },    
+}
+
+ML_MODEL_RESULTS = {
+    "coef_name": {
+        "model_results_key": OLS_MODEL_RESULTS["coef_name"]["model_results_key"],
+        "summary_description": OLS_MODEL_RESULTS["coef_name"]["summary_description"],
+    },
+    "coef": {
+        "model_results_key": OLS_MODEL_RESULTS["coef"]["model_results_key"],
+        "summary_description": OLS_MODEL_RESULTS["coef"]["summary_description"],
+    },
+    "coef_standard_errors": {
+        "model_results_key": OLS_MODEL_RESULTS["coef_standard_errors"]["model_results_key"],
+        "summary_description": OLS_MODEL_RESULTS["coef_standard_errors"]["summary_description"],
+    },
+    "coef_teststatistic": {
+        "model_results_key": "z",
+        "summary_description": "z"
+    },
+    "coef_p": {
+        "model_results_key": OLS_MODEL_RESULTS["coef_p"]["model_results_key"],
+        "summary_description": OLS_MODEL_RESULTS["coef_p"]["summary_description"]
+    },
+    "coef_confint_lower": {
+        "model_results_key": OLS_MODEL_RESULTS["coef_confint_lower"]["model_results_key"],
+        "summary_description": OLS_MODEL_RESULTS["coef_confint_lower"]["summary_description"]
+    },
+    "coef_confint_upper": {
+        "model_results_key": OLS_MODEL_RESULTS["coef_confint_upper"]["model_results_key"],
+        "summary_description": OLS_MODEL_RESULTS["coef_confint_upper"]["summary_description"]
+    },    
+}
+
+# Model fit metrics:
+
+MODEL_FIT_METRICS_DESCRIPTION = "Model fit metric"
+
+MODEL_FIT_METRICS = {
+    "SSR": {
+        "description": "Sum of squared residuals",
+        "show_in_summary": False
+        },
+    "SAR": {
+        "description": "Sum of absolute residuals",
+        "show_in_summary": False
+        },
+    "SQT": {
+        "description": "Total variance of dependent variable",
+        "show_in_summary": False
+        },
+    "RSQ": {
+        "description": "R-Squared",
+        "show_in_summary": True
+        },
+    "RSQ_ADJ": {
+        "description": "R-Squared adjusted",
+        "show_in_summary": True
+        },
+    "MSE": {
+        "description": "Mean squared error",
+        "show_in_summary": True
+        },
+    "RMSE": {
+        "description": "Root mean squared error",
+        "show_in_summary": True
+        },
+    "MAE": {
+        "description": "Mean absolute error",
+        "show_in_summary": True
+        },
+    "MAPE": {
+        "description": "Mean absolute percentage error",
+        "show_in_summary": True
+        },
+    "MAPE_SYM": {
+        "description": "Symmetric mean absolute percentage error",
+        "show_in_summary": False
+        }
+}
+
+# Treatment diagnostics:
+
+TREATMENT_DIAGNOSTICS_DESCRIPTION = f"{TREATMENT_DESCRIPTION} diagnostics"
+
+TREATMENT_DIAGNOSTICS = {
+    "treatment": {
+        "description": TREATMENT_DESCRIPTION,
+        "show_in_summary": True
+        },
+    "is_notreatment": {
+        "description": NO_TREATMENT_CG_DESCRIPTION,
+        "show_in_summary": True,
+        },
+    "treatment_group": {
+        "description": TREATMENT_GROUP_DESCRIPTION,
+        "show_in_summary": False
+        },
+    "control_group": {
+        "description": CONTROL_GROUP_DESCRIPTION,
+        "show_in_summary": False
+        },
+    "is_parallel": {
+        "description": "Parallel trends (pre)",
+        "show_in_summary": True
+        },
+    "is_simultaneous": {
+        "description": f"{TREATMENT_SIMULTANEOUS_DESCRIPTION} treatment",
+        "show_in_summary": False
+    },
+    "adoption_type": {
+        "description": "Type of adoption",
+        "show_in_summary": True
+    },
+    "is_binary": {
+        "description": "Binary treatment",
+        "show_in_summary": False
+    },
+    "treatment_format": {
+        "description": f"{TREATMENT_DESCRIPTION} format",
+        "show_in_summary": True
+    },
+    "treatment_group_size": {
+        "description": f"{TREATMENT_GROUP_DESCRIPTION} (N)",
+        "show_in_summary": True
+    },
+    "control_group_size": {
+        "description": f"{CONTROL_GROUP_DESCRIPTION} (N)",
+        "show_in_summary": True
+    },
+    "is_multiple_treatment_period": {
+        "description": "Multiple treatment periods",
+        "show_in_summary": True
+    },
+}
+
+# Input data diagnostics:
+
+DATA_DIAGNOSTICS_DESCRIPTION = "Input data diagnostics"
+
+DATA_DIAGNOSTICS = {
+    "is_balanced": {
+        "description": "Balanced panel data",
+        "show_in_summary": True
+        },
+    "is_missing": {
+        "description": "Missing values",
+        "show_in_summary": True
+        },
+    "drop_missing": {
+        "description": "Drop missing values",
+        "show_in_summary": False
+        }, 
+    "missing_replace_by_zero": {
+        "description": "Replace missing values by zero",
+        "show_in_summary": False
+        },
+    "is_prepost": {
+        "description": PREPOST_PANELDATA_DESCRIPTION,
+        "show_in_summary": False
+        },
+    "data_type": {
+        "description": "Data type",
+        "show_in_summary": True
+        },
+    "outcome_col": {
+        "description": "Outcome variable",
+        "show_in_summary": True
+        },
+    "outcome_descriptives": {
+        "description": "Outcome descriptives",
+        "show_in_summary": True
+        },
+    "observations": {
+        "description": N_DESCRIPTION,
+        "show_in_summary": True
+        },
+    }
+
+DIAGNOSTICS_COLUMN = "Result"
+
+COVARIATES_DESCRIPTION = "Covariates"
+
+# Predictions:
+
+PREDICTIONS_SUMMARY_FRAME_COLS = {
+    "mean": "Predicted mean",
+    "mean_se": "Predicted mean SE",
+    "mean_ci_lower": "Lower CI of mean",
+    "mean_ci_upper": "Upper CI of mean",
+    "obs_ci_lower": "Lower prediction interval",
+    "obs_ci_upper": "Upper prediction interval",
+}
+PREDICTIONS_SUMMARY_FRAME_COLS_LIST = list(PREDICTIONS_SUMMARY_FRAME_COLS.keys())
+PREDICTIONS_SUMMARY_FRAME_DESCRIPTIONS = list(PREDICTIONS_SUMMARY_FRAME_COLS.values())
+
+# Counterfactual:
+COUNTERFAC_SUFFIX_CF = "counterfac"
+COUNTERFAC_SUFFIX_PRED_CF = f"{DELIMITER}{PREDICTED_SUFFIX}{DELIMITER}{COUNTERFAC_SUFFIX_CF}"