diffindiff 2.5.0__tar.gz → 2.5.2__tar.gz

{diffindiff-2.5.0 → diffindiff-2.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: diffindiff
-Version: 2.5.0
+Version: 2.5.2
 Summary: diffindiff: Python library for convenient Difference-in-Differences analyses
 Author: Thomas Wieland
 Author-email: geowieland@googlemail.com
@@ -22,14 +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).
+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.5.0) [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.5.2) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
 
 
 ## Installation
@@ -53,6 +53,7 @@ pip install git+https://github.com/geowieland/diffindiff_official.git
   - Define custom treatment and control groups as well as treatment periods
   - Create ready-to-fit DiD data objects
   - Create predictive counterfactuals
+  - Create a synthetic control unit 
 - **DiD analysis**: 
   - Perform standard DiD analysis with pre-post data
   - Perform DiD analysis with two-way fixed effects models
@@ -66,6 +67,7 @@ pip install git+https://github.com/geowieland/diffindiff_official.git
     - Including after-treatment period
   - Perform Triple Difference (DDD) analysis
   - Perform DiD analysis with demeaned data
+  - Perform Synthetic DiD analysis
 - **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
@@ -177,16 +179,11 @@ 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.5.0)
+## What's new (v2.5.2)
 
-- General
-  - Update of dependencies to be compatible with Python 3.13 (still works with Python 3.11) while avoiding incompatibility conflict of statsmodels and scipy
-  - Internal changes in some functions to be compatible with Python 3.13 (still works with Python 3.11)
 - Extensions
-  - Creating synthetic control units with DiffData.add_synthetic() and conducting a Synthetic DiD analysis
-  - didanalysis_helper.treatment_diagnostics() now additionally checks unique number of analysis units and time points
-  - didanalysis_helper.data_diagnostics() now additionally checks whether covariates are constant; didanalysis.did_analysis() and .ddd_analysis() automatically skip such variables from the model analysis
+  - New functions DiffData.get_synthetic_control_weightings() and DiffData.get_synthetic_control_fitmetrics() for accessing control unit weightings and fit metrics of the synthetic control unit
 - Bugfixes
-  - DiffData.add_own_counterfactual() now works correctly in any case (also when no. of treatments > 1)
-  - Correct internal processing of treatment data in DiffTreatment and DiffData objects when treatments were added
-  - Fixed missing treatment group definition in DiffData.add_treatment() 
+  - DiffModel.placebo() now checks control group size and returns warning when no. of control units is equal to 1, not allowing any placebo test
+- Other
+  - Some updates of documentation and README  

{diffindiff-2.5.0 → diffindiff-2.5.2}/README.md

@@ -14,14 +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).
+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.5.0) [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.5.2) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
 
 
 ## Installation
@@ -45,6 +45,7 @@ pip install git+https://github.com/geowieland/diffindiff_official.git
   - Define custom treatment and control groups as well as treatment periods
   - Create ready-to-fit DiD data objects
   - Create predictive counterfactuals
+  - Create a synthetic control unit 
 - **DiD analysis**: 
   - Perform standard DiD analysis with pre-post data
   - Perform DiD analysis with two-way fixed effects models
@@ -58,6 +59,7 @@ pip install git+https://github.com/geowieland/diffindiff_official.git
     - Including after-treatment period
   - Perform Triple Difference (DDD) analysis
   - Perform DiD analysis with demeaned data
+  - Perform Synthetic DiD analysis
 - **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
@@ -169,16 +171,11 @@ 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.5.0)
+## What's new (v2.5.2)
 
-- General
-  - Update of dependencies to be compatible with Python 3.13 (still works with Python 3.11) while avoiding incompatibility conflict of statsmodels and scipy
-  - Internal changes in some functions to be compatible with Python 3.13 (still works with Python 3.11)
 - Extensions
-  - Creating synthetic control units with DiffData.add_synthetic() and conducting a Synthetic DiD analysis
-  - didanalysis_helper.treatment_diagnostics() now additionally checks unique number of analysis units and time points
-  - didanalysis_helper.data_diagnostics() now additionally checks whether covariates are constant; didanalysis.did_analysis() and .ddd_analysis() automatically skip such variables from the model analysis
+  - New functions DiffData.get_synthetic_control_weightings() and DiffData.get_synthetic_control_fitmetrics() for accessing control unit weightings and fit metrics of the synthetic control unit
 - Bugfixes
-  - DiffData.add_own_counterfactual() now works correctly in any case (also when no. of treatments > 1)
-  - Correct internal processing of treatment data in DiffTreatment and DiffData objects when treatments were added
-  - Fixed missing treatment group definition in DiffData.add_treatment() 
+  - DiffModel.placebo() now checks control group size and returns warning when no. of control units is equal to 1, not allowing any placebo test
+- Other
+  - Some updates of documentation and README  

{diffindiff-2.5.0 → diffindiff-2.5.2}/diffindiff/config.py

@@ -4,15 +4,15 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     1.0.21
-# Last update: 2026-07-03 17:50
+# Version:     1.0.23
+# Last update: 2026-07-14 19:48
 # Copyright (c) 2025-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 
 # Basic config:
 
 PACKAGE_NAME = "diffindiff"
-PACKAGE_VERSION = "2.5.0"
+PACKAGE_VERSION = "2.5.2"
 
 VERBOSE = False
 
@@ -52,6 +52,7 @@ TREATMENT_PERIOD_DESCRIPTION = f"{TREATMENT_DESCRIPTION} period"
 STUDY_PERIOD_DESCRIPTION = "Study period"
 PREPOST_DESCRIPTION = "Pre-post"
 AFTER_TREATMENT_PERIOD_DESCRIPTION = "After-treatment period"
+FOLLOW_UP_DESCRIPTION = "Follow-up"
 
 UNITS_DESCRIPTION = "Units"
 

{diffindiff-2.5.0 → diffindiff-2.5.2}/diffindiff/didanalysis.py

@@ -4,8 +4,8 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     2.4.1
-# Last update: 2026-07-02 20:17
+# Version:     2.4.3
+# Last update: 2026-07-14 19:49
 # Copyright (c) 2024-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 
@@ -898,6 +898,49 @@ class DiffModel:
         
         treatment_effects = self.treatment_effects()
 
+        try:
+            if model_config.get("pre_post", False) and len(model_config.get("after_treatment_col", [])) > 0:
+
+                labels = treatment_effects.iloc[:, 0].astype(str).values
+                has_after = any((config.AFTER_TREATMENT_PERIOD_DESCRIPTION in lab) or (config.ATT_COL in lab) for lab in labels)
+                if not has_after:
+                    model_results = self.data[0]
+
+                    aate_key = config.EFFECTS_TYPES["AATE"]["model_results_key"] if "AATE" in config.EFFECTS_TYPES else "average_after_treatment_effects"
+                    att_key = config.EFFECTS_TYPES["ATT"]["model_results_key"] if "ATT" in config.EFFECTS_TYPES else "after_treatment_time_effects"
+                    appended = False
+
+                    for key_try in [aate_key, att_key]:
+                        if key_try in model_results and isinstance(model_results[key_try], dict):
+                            for subkey, val in model_results[key_try].items():
+
+                                if isinstance(val, dict):
+
+                                    coef_k = config.OLS_MODEL_RESULTS["coef"]["model_results_key"]
+                                    se_k = config.OLS_MODEL_RESULTS["coef_standard_errors"]["model_results_key"]
+                                    t_k = config.OLS_MODEL_RESULTS["coef_teststatistic"]["model_results_key"]
+                                    p_k = config.OLS_MODEL_RESULTS["coef_p"]["model_results_key"]
+                                    cil_k = config.OLS_MODEL_RESULTS["coef_confint_lower"]["model_results_key"]
+                                    ciu_k = config.OLS_MODEL_RESULTS["coef_confint_upper"]["model_results_key"]
+
+                                    new_row = {
+                                        "": f"{config.EFFECTS_TYPES.get('AATE', {'description': 'After-treatment'})['description']} {subkey}",
+                                        "Estimate": val.get(coef_k, np.nan),
+                                        "SE": val.get(se_k, np.nan),
+                                        "t": val.get(t_k, np.nan),
+                                        "p": val.get(p_k, np.nan),
+                                        "CI lower": val.get(cil_k, np.nan),
+                                        "CI upper": val.get(ciu_k, np.nan),
+                                    }
+                                    treatment_effects = pd.concat([treatment_effects, pd.DataFrame([new_row])], ignore_index=True)
+                                    appended = True
+                                    break
+                        if appended:
+                            break
+        except Exception:
+
+            pass
+
         if sort_by_coef:
             treatment_effects = treatment_effects.sort_values(
                 by = treatment_effects.columns[1],
@@ -1379,10 +1422,15 @@ class DiffModel:
         model_data_c[time_col] = pd.to_datetime(model_data_c[time_col])
         model_data_c[unit_col] = model_data_c[unit_col].astype(str)
 
-        units_random_sample = model_data_c[unit_col].sample(
-            n = int(round(divide*control_group_N*resample, 0)), 
-            random_state = random_state
-            ).astype(str).tolist()
+        if control_group_N < 2:
+            print(f"WARNING: Model includes one control unit only: '{control_group[0]}'. Placebo analysis requires size of control group >1. Original model is returned.")
+            return self
+
+        else:
+            units_random_sample = model_data_c[unit_col].sample(
+                n = int(round(divide*control_group_N*resample, 0)), 
+                random_state = random_state
+                ).astype(str).tolist()
 
         model_data_c[TG_col_] = 0
         model_data_c.loc[(model_data_c[unit_col].isin(units_random_sample)), TG_col_] = 1
@@ -1807,14 +1855,14 @@ class DiffModel:
         model_data_TG_CG["t"] = pd.to_datetime(model_data_TG_CG["t"])
 
         if not model_config["pre_post"]:
-            pre_post_barplot = False        
+            pre_post_barplot = False
 
         if pre_post_barplot:
 
             x_pos_t1_TG = 0
             x_pos_t1_CG = x_pos_t1_TG + pre_post_bar_width  
             x_pos_t2_TG = 1.5  
-            x_pos_t2_CG = x_pos_t2_TG + pre_post_bar_width  
+            x_pos_t2_CG = x_pos_t2_TG + pre_post_bar_width
 
             plt.bar(
                 x = x_pos_t1_TG, 
@@ -1842,6 +1890,24 @@ class DiffModel:
                 color=lines_col[3], 
                 width = pre_post_bar_width
                 )
+            
+            if len(model_data_TG_CG) == 3:
+
+                x_pos_t3_TG = 3  
+                x_pos_t3_CG = x_pos_t3_TG + pre_post_bar_width
+
+                plt.bar(
+                    x = x_pos_t3_TG, 
+                    height = model_data_TG_CG[outcome_col_expected_TG][2],                 
+                    color = lines_col[2], 
+                    width = pre_post_bar_width
+                    )            
+                plt.bar(
+                    x = x_pos_t3_CG, 
+                    height = model_data_TG_CG[outcome_col_expected_CG][2],                 
+                    color=lines_col[3], 
+                    width = pre_post_bar_width
+                    )
 
             plt.xlabel(x_label)
             plt.ylabel(y_label)
@@ -1935,16 +2001,29 @@ class DiffModel:
             ax.xaxis.set_major_formatter(DateFormatter(model_config["date_format"]))
 
         if model_config["pre_post"]:
+
+            ticks_line = model_data_TG_CG["t"].unique()
+            ticks_bars = [0.25, 1.75]
+
+            if len(model_data_TG_CG) == 3:
+                
+                if len(pre_post_ticks) < 3:
+                    pre_post_ticks.append("Follow-up")
+
+                ticks_bars.append(3.25)
+
             if not pre_post_barplot:
+                
                 plt.xticks(
-                    model_data_TG_CG["t"].unique(), 
-                    labels = [pre_post_ticks[0], pre_post_ticks[1]]
+                    ticks_line, 
+                    labels = pre_post_ticks
                     )  
             else:
                 plt.xticks(
-                    [0.25, 1.75], 
-                    labels = [pre_post_ticks[0], pre_post_ticks[1]]
-                    )  
+                    ticks_bars, 
+                    labels = pre_post_ticks
+                    )
+                
         else:
             plt.xticks(rotation=90)
         
@@ -2644,6 +2723,8 @@ def did_analysis(
     FE_time_vars = []
     dummy_time_original = []
     
+    drop_first = intercept
+
     if not demean:
 
         if FE_unit:
@@ -2652,7 +2733,7 @@ def did_analysis(
                 data = data,
                 col = unit_col,
                 type = "unit",
-                drop_first = intercept,
+                drop_first = drop_first,
                 verbose = verbose
             )
             
@@ -2667,7 +2748,7 @@ def did_analysis(
                 data = data,
                 col = time_col,
                 type = "time",
-                drop_first = intercept,
+                drop_first = drop_first,
                 verbose = verbose
             )
             
@@ -2691,7 +2772,7 @@ def did_analysis(
                 data = data,
                 col = group_by,
                 type = "group",
-                drop_first = intercept,
+                drop_first = drop_first,
                 verbose = verbose
                 )
         

{diffindiff-2.5.0 → diffindiff-2.5.2}/diffindiff/didanalysis_helper.py

@@ -4,8 +4,8 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     1.2.2
-# Last update: 2026-07-02 20:18
+# Version:     1.2.3
+# Last update: 2026-07-14 19:44
 # Copyright (c) 2025-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 

{diffindiff-2.5.0 → diffindiff-2.5.2}/diffindiff/diddata.py

@@ -4,8 +4,8 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     2.3.0
-# Last update: 2026-07-03 18:00
+# Version:     2.3.2
+# Last update: 2026-07-16 18:52
 # Copyright (c) 2024-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 
@@ -622,15 +622,36 @@ def create_treatment(
 
     if pre_post:
 
-        after_treatment_period = False
+        if after_treatment_period:
+
+            if study_period[1] <= treatment_period[1]:
+                
+                print(f"WARNING: after_treatment_period is set True, but end of study period ({study_period[1]}) is <= end of treatment period ({treatment_period[1]}). After treatment period is ignored.")
+                
+                after_treatment_period = False
 
-        study_period_range = [study_period[0], study_period[1]]
-        study_period_N = 2
-        study_period_counter = [1,2]
+            else:
+
+                study_period_range = [study_period[0], study_period[1]]
+                study_period_N = 3
+                study_period_counter = [1,2,3]
 
-        treatment_period_range = [treatment_period[0], treatment_period[1]]
-        treatment_period_N = 1
-        TT_dummies = [0,1]
+                treatment_period_N = 1
+                TT_dummies = [0,1,0]
+                ATT_dummies = [0,0,1]
+
+                treatment_period_range = [study_period[0], treatment_period[1], study_period[1]]
+                after_treatment_period_range = treatment_period_range
+        
+        else:
+                
+            study_period_range = [study_period[0], study_period[1]]
+            study_period_N = 2
+            study_period_counter = [1,2]
+
+            treatment_period_range = [treatment_period[0], treatment_period[1]]
+            treatment_period_N = 1
+            TT_dummies = [0,1]
 
         study_period_range = pd.DataFrame (treatment_period_range, columns=[config.TIME_COL])
         study_period_range[config.TIME_COUNTER_COL] = pd.DataFrame(study_period_counter)
@@ -640,7 +661,7 @@ def create_treatment(
             TT_col: TT_dummies
             }
         
-        TT_data = pd.DataFrame(TT_data)
+        TT_data = pd.DataFrame(TT_data)    
         
         treatment_period_range = pd.DataFrame(
             study_period_range
@@ -650,6 +671,24 @@ def create_treatment(
             TT_data, 
             how = "left"
             )
+        
+        if after_treatment_period:
+
+            ATT_data = {
+                config.TIME_COL: after_treatment_period_range, 
+                ATT_col: ATT_dummies
+                }
+
+            ATT_data = pd.DataFrame(ATT_data)
+
+            after_treatment_period_range = pd.DataFrame (after_treatment_period_range, columns=[config.TIME_COL])
+            after_treatment_period_range[config.TIME_COUNTER_COL] = pd.DataFrame(study_period_counter)
+
+            after_treatment_data_df = after_treatment_period_range.merge(ATT_data, how = "left")
+            after_treatment_data_df[ATT_col] = after_treatment_data_df[ATT_col].fillna(0)
+            after_treatment_data_df = after_treatment_data_df.drop(columns=[config.TIME_COL, config.TIME_COUNTER_COL])
+
+            treatment_data_df = pd.concat([treatment_data_df, after_treatment_data_df], axis=1)
 
     else:
 
@@ -689,7 +728,7 @@ def create_treatment(
 
     treatment_data_df[TT_col] = treatment_data_df[TT_col].fillna(0)
 
-    if after_treatment_period:
+    if after_treatment_period and not pre_post:
         
         treatment_period_last = datetime.strptime(
             treatment_period[1], 
@@ -720,6 +759,9 @@ def create_treatment(
 
     else:
         after_treatment_period_N = 0
+
+    if after_treatment_period and pre_post:
+        after_treatment_period_N = 1
     
     no_treatments = 1
 
@@ -735,7 +777,7 @@ def create_treatment(
         "date_format": date_format,
         "pre_post": pre_post
         }
-
+    
     treatment_config = {
         0: {
             "treatment_name": treatment_name,
@@ -773,7 +815,8 @@ class DiffData:
         unit_time_col_original,
         covariates,
         treatment_cols,
-        timestamp
+        timestamp,
+        synthetic_control
         ):
 
         """
@@ -798,6 +841,8 @@ class DiffData:
             Mapping of treatment columns metadata.
         timestamp : dict
             Timestamp metadata.
+        synthetic_control : dict
+            If included, data for a synthetic control unit, else None.
         Returns
         -------
         None
@@ -813,7 +858,8 @@ class DiffData:
             unit_time_col_original,
             covariates,
             treatment_cols,
-            timestamp
+            timestamp,
+            synthetic_control
             ]
 
     def get_did_modeldata_df (self):
@@ -1018,6 +1064,46 @@ class DiffData:
         """
 
         return self.data[7]
+    
+    def get_synthetic_control_weightings(self):
+
+        """
+        Return weightings of the synthetic control unit (if included in the DiffData object).
+
+        Returns
+        -------
+        pandas.DataFrame
+            Weighting by control unit.
+        
+        Examples
+        --------
+        >>> my_diffdata.get_synthetic_control_weightings()
+        """
+
+        if len(self.data[8]) > 0:
+            return self.data[8]["weights"]
+        else:
+            print("The DiffData object does not include a synthetic control unit.")
+            
+    def get_synthetic_control_fitmetrics(self):
+
+        """
+        Return fit metrics of the synthetic control unit (if included in the DiffData object).
+
+        Returns
+        -------
+        list
+            Fit metrics (dict) and data points with residuals (pandas.DataFrame).
+        
+        Examples
+        --------
+        >>> my_diffdata.get_synthetic_control_fitmetrics()
+        """
+
+        if len(self.data[8]) > 0:
+            return self.data[8]["fit_metrics"]
+        else:
+            print("The DiffData object does not include a synthetic control unit.")
 
     def add_covariates(
         self, 
@@ -1991,6 +2077,14 @@ class DiffData:
             verbose = False
             )
         
+        self.data[7] = helper.create_timestamp(function="add_synthetic")
+        
+        self.data[8] = {
+            "weights": weights_pd,
+            "fit_metrics": synthetic_fit_metrics,
+            "process_unit": process_unit
+        }
+        
         return self
  
     def summary(self):
@@ -2050,7 +2144,10 @@ class DiffData:
             else:
                 print(f"  {config.STUDY_PERIOD_DESCRIPTION:<{config.DIDDATA_SUMMARY_MAX_WIDTH}} {treatment_meta['study_period_start']} - {treatment_meta['study_period_end']} ({treatment_meta['study_period']} {treatment_meta['frequency']})")
                 print(f"  {config.TREATMENT_PERIOD_DESCRIPTION:<{config.DIDDATA_SUMMARY_MAX_WIDTH}} {treatment_config[key]['treatment_period_start']} - {treatment_config[key]['treatment_period_end']} ({treatment_config[key]['treatment_period']} {treatment_meta['frequency']})")
-                
+
+            if value["after_treatment_period"]:
+                print(f"  {config.AFTER_TREATMENT_PERIOD_DESCRIPTION:<{config.DIDDATA_SUMMARY_MAX_WIDTH}} {treatment_config[key]['treatment_period_end']} vs. {treatment_meta['study_period_end']} ({config.FOLLOW_UP_DESCRIPTION})")
+
         print("-" * total_width)
 
         print(f"Outcome '{outcome_col_original}'")
@@ -2444,7 +2541,8 @@ def merge_data(
         unit_time_col_original,
         [],
         treatment_cols,
-        timestamp
+        timestamp,
+        {}
         )
 
     if verbose:

{diffindiff-2.5.0 → diffindiff-2.5.2}/diffindiff/tests/tests_diffindiff.py

@@ -4,8 +4,8 @@
 # Author:      Thomas Wieland 
 #              ORCID: 0000-0001-5168-9846
 #              mail: geowieland@googlemail.com              
-# Version:     2.0.15
-# Last update: 2026-05-01 09:54
+# Version:     2.1.0
+# Last update: 2026-07-10 11:35
 # Copyright (c) 2025-2026 Thomas Wieland
 #-----------------------------------------------------------------------
 
@@ -75,7 +75,7 @@ curfew_data_prepost=create_data(
     treatment_period=["2020-03-21", "2020-05-05"],
     freq="D",
     pre_post=True,
-    after_treatment_period = True
+    after_treatment_period = False
     )
 # Creating DiD treatement dataset by defining groups and
 # treatment time at once
@@ -98,6 +98,9 @@ print(curfew_model_prepost.covariates())
 curfew_model_prepost.summary()
 # Model summary
 
+print(curfew_model_prepost.didmodel().summary())
+# Access statsmodels OLS object
+
 print(curfew_model_prepost.fit_metrics())
 # Show model fit metrics
 
@@ -137,6 +140,59 @@ curfew_model_prepost.plot_treatment_effects(
     )
 # plot effects
 
+
+# Same analysis with after-treatment period (follow-up):
+
+curfew_data_prepost_AT=create_data(
+    outcome_data=curfew_DE,
+    unit_id_col="county",
+    time_col="infection_date",
+    outcome_col="infections_cum_per100000",
+    treatment_group= 
+        curfew_DE.loc[curfew_DE["Bundesland"].isin([9,10,14])]["county"],
+    control_group= 
+        curfew_DE.loc[~curfew_DE["Bundesland"].isin([9,10,14])]["county"],
+    study_period=["2020-03-01", "2020-05-15"],
+    treatment_period=["2020-03-21", "2020-05-05"],
+    freq="D",
+    pre_post=True,
+    after_treatment_period = True
+    )
+# Creating DiD treatement dataset by defining groups and
+# treatment time at once
+
+curfew_data_prepost_AT.summary()
+# Summary of created data
+
+curfew_model_prepost_AT=curfew_data_prepost_AT.analysis(verbose=True)
+# Model analysis of created data
+
+curfew_model_prepost_AT.summary()
+# Model summary
+
+curfew_model_prepost_AT.plot(
+    x_label="Timepoint",
+    y_label="Cumulative infections per 100,000",
+    plot_title="Curfew effectiveness pre-post - Groups over time",
+    plot_observed=False,
+    lines_col=[None,None,"blue","orange"],
+    lines_labels=[None,None,"Treatment group","Control group","Treatment group CI","Control group CI"],
+    lines_style=[None,None,"solid","solid"]
+    )
+# Plot DiD pre vs. post results
+# with user-determined style
+
+curfew_model_prepost_AT.plot(
+    x_label="Timepoint",
+    y_label="Cumulative infections per 100,000",
+    plot_title="Curfew effectiveness pre-post - Groups over time",
+    lines_col=[None,None,"blue","orange"],
+    lines_labels=[None,None,"Treatment group","Control group","Treatment group CI", "Control group CI"],
+    pre_post_barplot=True
+    )
+# Plot DiD pre vs. post results
+# with user-determined style
+
 counties_DE=pd.read_csv("data/counties_DE.csv", sep=";", decimal=",", encoding='latin1')
 # Dataset with German county data
 

{diffindiff-2.5.0 → diffindiff-2.5.2}/diffindiff.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: diffindiff
-Version: 2.5.0
+Version: 2.5.2
 Summary: diffindiff: Python library for convenient Difference-in-Differences analyses
 Author: Thomas Wieland
 Author-email: geowieland@googlemail.com
@@ -22,14 +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).
+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.5.0) [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.5.2) [Computer software]. Zenodo. https://doi.org/10.5281/zenodo.18656820
 
 
 ## Installation
@@ -53,6 +53,7 @@ pip install git+https://github.com/geowieland/diffindiff_official.git
   - Define custom treatment and control groups as well as treatment periods
   - Create ready-to-fit DiD data objects
   - Create predictive counterfactuals
+  - Create a synthetic control unit 
 - **DiD analysis**: 
   - Perform standard DiD analysis with pre-post data
   - Perform DiD analysis with two-way fixed effects models
@@ -66,6 +67,7 @@ pip install git+https://github.com/geowieland/diffindiff_official.git
     - Including after-treatment period
   - Perform Triple Difference (DDD) analysis
   - Perform DiD analysis with demeaned data
+  - Perform Synthetic DiD analysis
 - **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
@@ -177,16 +179,11 @@ 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.5.0)
+## What's new (v2.5.2)
 
-- General
-  - Update of dependencies to be compatible with Python 3.13 (still works with Python 3.11) while avoiding incompatibility conflict of statsmodels and scipy
-  - Internal changes in some functions to be compatible with Python 3.13 (still works with Python 3.11)
 - Extensions
-  - Creating synthetic control units with DiffData.add_synthetic() and conducting a Synthetic DiD analysis
-  - didanalysis_helper.treatment_diagnostics() now additionally checks unique number of analysis units and time points
-  - didanalysis_helper.data_diagnostics() now additionally checks whether covariates are constant; didanalysis.did_analysis() and .ddd_analysis() automatically skip such variables from the model analysis
+  - New functions DiffData.get_synthetic_control_weightings() and DiffData.get_synthetic_control_fitmetrics() for accessing control unit weightings and fit metrics of the synthetic control unit
 - Bugfixes
-  - DiffData.add_own_counterfactual() now works correctly in any case (also when no. of treatments > 1)
-  - Correct internal processing of treatment data in DiffTreatment and DiffData objects when treatments were added
-  - Fixed missing treatment group definition in DiffData.add_treatment() 
+  - DiffModel.placebo() now checks control group size and returns warning when no. of control units is equal to 1, not allowing any placebo test
+- Other
+  - Some updates of documentation and README  

{diffindiff-2.5.0 → diffindiff-2.5.2}/setup.py

@@ -7,7 +7,7 @@ def read_README():
     
 setup(
     name='diffindiff',
-    version='2.5.0',
+    version='2.5.2',
     description='diffindiff: Python library for convenient Difference-in-Differences analyses',
     packages=find_packages(include=["diffindiff", "diffindiff.tests"]),
     include_package_data=True,