diffindiff 2.0.0__tar.gz → 2.0.1__tar.gz

{diffindiff-2.0.0 → diffindiff-2.0.1}/PKG-INFO

@@ -1,10 +1,24 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: diffindiff
-Version: 2.0.0
+Version: 2.0.1
 Summary: diffindiff: Python library for convenient Difference-in-Differences Analyses
 Author: Thomas Wieland
 Author-email: geowieland@googlemail.com
 Description-Content-Type: text/markdown
+Requires-Dist: numpy
+Requires-Dist: pandas
+Requires-Dist: statsmodels
+Requires-Dist: matplotlib
+Requires-Dist: datetime
+Requires-Dist: scikit-learn
+Requires-Dist: xgboost
+Requires-Dist: lightgbm
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-dist
+Dynamic: summary
 
 # diffindiff: Difference-in-Differences (DiD) Analysis Python Library
 

{diffindiff-2.0.0 → diffindiff-2.0.1}/diffindiff/didanalysis.py

@@ -1,9 +1,9 @@
 #-------------------------------------------------------------------------------
 # Name:        didanalysis (diffindiff)
 # Purpose:     Analysis functions for difference-in-differences analyses
-# Author:      Thomas Wieland (geowieland@googlemail.com)
-# Version:     2.0.0
-# Last update: 2025-04-15 07:11
+# Author:      Thomas Wieland (mail: geowieland@googlemail.com, ORCID: 0000-0001-5168-9846)
+# Version:     2.0.1
+# Last update: 2025-04-15 18:43
 # Copyright (c) 2025 Thomas Wieland
 #-------------------------------------------------------------------------------
 
@@ -13,7 +13,7 @@ from statsmodels.formula.api import ols
 import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.dates import DateFormatter
-import didtools
+import diffindiff.didtools
 
 
 class DiffModel:
@@ -686,7 +686,7 @@ class DiffModel:
         model_data = self.data[2]
         model_config = self.data[1]
 
-        modeldata_isparallel = didtools.is_parallel(
+        modeldata_isparallel = diffindiff.didtools.is_parallel(
             data = model_data,
             unit_col = model_config["unit_col"],
             time_col = model_config["time_col"],
@@ -1275,7 +1275,7 @@ def did_analysis(
     missing_replace_by_zero: bool = False
     ):
     
-    didtools.check_columns(
+    diffindiff.didtools.check_columns(
         df = data,
         columns = [unit_col, time_col, outcome_col]
         )
@@ -1287,7 +1287,7 @@ def did_analysis(
     no_treatments = len(treatment_col)
     if no_treatments == 0:
         raise ValueError ("No treatment(s) in treatment_col stated.")    
-    didtools.check_columns(
+    diffindiff.didtools.check_columns(
         df = data,
         columns = treatment_col
         )    
@@ -1301,12 +1301,12 @@ def did_analysis(
     treatment_diagnostics = {}    
     staggered_adoption = False
     for i, treatment in enumerate(treatment_col):    
-        is_notreatment = didtools.is_notreatment(
+        is_notreatment = diffindiff.didtools.is_notreatment(
             data = data,
             unit_col = unit_col,
             treatment_col = treatment
             )        
-        is_parallel = didtools.is_parallel(
+        is_parallel = diffindiff.didtools.is_parallel(
             data = data,
             unit_col = unit_col,
             time_col = time_col,
@@ -1315,13 +1315,13 @@ def did_analysis(
             pre_post = pre_post,
             alpha = confint_alpha
             )        
-        is_simultaneous = didtools.is_simultaneous(
+        is_simultaneous = diffindiff.didtools.is_simultaneous(
             data = data,
             unit_col = unit_col,
             time_col = time_col,
             treatment_col = treatment
             )        
-        is_binary = didtools.is_binary(
+        is_binary = diffindiff.didtools.is_binary(
             data = data,
             treatment_col = treatment
             )    
@@ -1363,7 +1363,7 @@ def did_analysis(
         if isinstance (after_treatment_col, str):
             after_treatment_col = [after_treatment_col]
         after_treatment_col = [entry for entry in after_treatment_col if entry is not None]
-        didtools.check_columns(
+        diffindiff.didtools.check_columns(
             df = data,
             columns = after_treatment_col
             )        
@@ -1373,7 +1373,7 @@ def did_analysis(
         if isinstance (TG_col, str):
             TG_col = [TG_col]
         TG_col = [entry for entry in TG_col if entry is not None]        
-        didtools.check_columns(
+        diffindiff.didtools.check_columns(
             df = data,
             columns = TG_col
             )        
@@ -1385,7 +1385,7 @@ def did_analysis(
         if isinstance (TT_col, str):
             TT_col = [TT_col]
         TT_col = [entry for entry in TT_col if entry is not None]
-        didtools.check_columns(
+        diffindiff.didtools.check_columns(
             df = data,
             columns = TT_col
             )        
@@ -1394,14 +1394,14 @@ def did_analysis(
         FE_time = True
 
     if covariates is not None or (isinstance (covariates, list) and len(covariates) > 0):
-        didtools.check_columns(
+        diffindiff.didtools.check_columns(
             df = data,
             columns = covariates
             )        
         cols_relevant = cols_relevant + covariates
 
     if group_by is not None and group_by != "":
-        didtools.check_columns(
+        diffindiff.didtools.check_columns(
             df = data,
             columns = [group_by]
             )
@@ -1412,13 +1412,13 @@ def did_analysis(
     data = data[cols_relevant].copy()
     
     if "date_counter" not in data.columns:    
-        data = didtools.date_counter(
+        data = diffindiff.didtools.date_counter(
             data,
             time_col,
             new_col = "date_counter"
             )
 
-    modeldata_ismissing = didtools.is_missing(
+    modeldata_ismissing = diffindiff.didtools.is_missing(
         data, 
         drop_missing = drop_missing,
         missing_replace_by_zero = missing_replace_by_zero
@@ -1435,7 +1435,7 @@ def did_analysis(
             print ("Missing values are not cleaned. Model may crash.")   
     
     other_cols_relevant = [col for col in cols_relevant if col not in [unit_col, time_col, outcome_col]]
-    modeldata_isbalanced = didtools.is_balanced(
+    modeldata_isbalanced = diffindiff.didtools.is_balanced(
         data = data,
         unit_col = unit_col,
         time_col = time_col,
@@ -1474,7 +1474,7 @@ def did_analysis(
         did_formula = did_formula + f'+ {" + ".join(after_treatment_col)}'
     
     if FE_unit:
-        unit_col_todummies = didtools.to_dummies(
+        unit_col_todummies = diffindiff.didtools.to_dummies(
             data = data,
             col = unit_col,
             prefix = "UNIT",
@@ -1486,7 +1486,7 @@ def did_analysis(
         dummy_unit_original = list(unit_col_todummies[2][unit_col].values)
 
     if FE_time:
-        time_col_todummies = didtools.to_dummies(   
+        time_col_todummies = diffindiff.didtools.to_dummies(   
             data = data,
             col = time_col,
             prefix = "TIME",
@@ -1501,7 +1501,7 @@ def did_analysis(
         if group_by is None or group_by == "":
             print ("WARNING: Grouping variable is not defined. No group-specific analyses are carried out. Define a grouping variable using group_by.")
         else:            
-            group_col_todummies = didtools.to_dummies(
+            group_col_todummies = diffindiff.didtools.to_dummies(
                 data = data,
                 col = group_by,
                 prefix = "GROUP",
@@ -1515,7 +1515,7 @@ def did_analysis(
     if GTT:        
         if group_by is not None and group_by != "":
             if "date_counter" not in data.columns:
-                data = didtools.date_counter(
+                data = diffindiff.didtools.date_counter(
                     data,
                     time_col,
                     new_col="date_counter"
@@ -1531,7 +1531,7 @@ def did_analysis(
     
     if ITT:        
         if "date_counter" not in data.columns:
-            data = didtools.date_counter(
+            data = diffindiff.didtools.date_counter(
                 data,
                 time_col,
                 new_col="date_counter"
@@ -1577,7 +1577,7 @@ def did_analysis(
         did_formula = did_formula + f'+{covariates_join}'
 
     if len(group_benefit) > 0:        
-        group_benefit = didtools.unique(group_benefit)
+        group_benefit = diffindiff.didtools.unique(group_benefit)
         if no_treatments == 1:       
             DDD = True
             if "TG_"+treatment_diagnostics[0]["treatment"] not in data.columns:                

{diffindiff-2.0.0 → diffindiff-2.0.1}/diffindiff/diddata.py

@@ -1,9 +1,9 @@
 #-------------------------------------------------------------------------------
 # Name:        diddata (diffindiff)
 # Purpose:     Creating data for Difference-in-Differences Analysis
-# Author:      Thomas Wieland (geowieland@googlemail.com)
-# Version:     2.0.0
-# Last update: 2025-04-15 07:21
+# Author:      Thomas Wieland (mail: geowieland@googlemail.com, ORCID: 0000-0001-5168-9846)
+# Version:     2.0.1
+# Last update: 2025-04-15 18:43
 # Copyright (c) 2025 Thomas Wieland
 #-------------------------------------------------------------------------------
 
@@ -11,8 +11,8 @@ import pandas as pd
 import numpy as np
 from datetime import datetime, timedelta
 import re
-import didanalysis
-import didtools
+import diffindiff.didanalysis
+import diffindiff.didtools
 
 
 class DiffGroups:
@@ -80,8 +80,8 @@ def create_groups(
     if treatment_name is not None:
         TG_col = "TG_"+treatment_name
 
-    treatment_group_unique = didtools.unique(treatment_group)
-    control_group_unique = didtools.unique(control_group)
+    treatment_group_unique = diffindiff.didtools.unique(treatment_group)
+    control_group_unique = diffindiff.didtools.unique(control_group)
 
     treatment_group_N = len(treatment_group_unique)
     control_group_N = len(control_group_unique)
@@ -578,7 +578,7 @@ class DiffData:
         if time_col is None or counterfactual_outcome_col is None:
             raise ValueError("time_col and counterfactual_outcome_col must be stated")
         
-        didtools.check_columns(
+        diffindiff.didtools.check_columns(
             df = additional_df,
             columns = [counterfactual_outcome_col, time_col]
             )
@@ -730,7 +730,7 @@ class DiffData:
         if groups_config[0]["DDD"]:
             group_benefit = did_modeldata.loc[did_modeldata["group_benefit"] == 1, "unit_UID"].unique()            
 
-        did_results = didanalysis.did_analysis(
+        did_results = diffindiff.didanalysis.did_analysis(
             data = did_modeldata,
             TG_col = TG_col,
             TT_col = TT_col,
@@ -810,7 +810,7 @@ def merge_data(
         )
     
     if drop_missing or missing_replace_by_zero:
-        modeldata_ismissing = didtools.is_missing(
+        modeldata_ismissing = diffindiff.didtools.is_missing(
             data = did_modeldata, 
             drop_missing = drop_missing,
             missing_replace_by_zero = missing_replace_by_zero
@@ -943,7 +943,7 @@ def create_counterfactual(
     data = data[data[unit_col].astype(str) != cf_for_unit]
     data_unit = data[data[unit_col].astype(str) == cf_for_unit]
     
-    isnotreatment = didtools.is_notreatment(
+    isnotreatment = diffindiff.didtools.is_notreatment(
         data = data,
         unit_col = unit_col,
         treatment_col = treatment_col
@@ -956,7 +956,7 @@ def create_counterfactual(
         time_col = time_col,
         treatment_col = treatment_col
         )
-    units = didtools.unique(units_tt[unit_col])  
+    units = diffindiff.didtools.unique(units_tt[unit_col])  
     
     if not isnotreatment[0]:
         print ("No no-treatment control group. Counterfactual will not cover full treatment time.")    
@@ -981,7 +981,7 @@ def create_counterfactual(
             ignore_index=True
         )
         
-    counterfactual_pred = didtools.model_wrapper(
+    counterfactual_pred = diffindiff.didtools.model_wrapper(
         y = data_cf[y],
         X = data_cf[X],
         model_type = model_type,

{diffindiff-2.0.0 → diffindiff-2.0.1}/diffindiff/didtools.py

@@ -1,9 +1,9 @@
 # -------------------------------------------------------------------------------
 # Name:        didtools (diffindiff)
 # Purpose:     Creating data for Difference-in-Differences Analysis
-# Author:      Thomas Wieland (geowieland@googlemail.com)
-# Version:     2.0.0
-# Last update: 2025-04-15 07:05
+# Author:      Thomas Wieland (mail: geowieland@googlemail.com, ORCID: 0000-0001-5168-9846)
+# Version:     2.0.1
+# Last update: 2025-04-15 18:44
 # Copyright (c) 2025 Thomas Wieland
 #-------------------------------------------------------------------------------
 

{diffindiff-2.0.0 → diffindiff-2.0.1}/diffindiff/tests/tests_diffindiff.py

@@ -3,8 +3,8 @@
 # Name:        tests_diffindiff
 # Purpose:     Tests and examples for the diffindiff package
 # Author:      Thomas Wieland (mail: geowieland@googlemail.com, ORCID: 0000-0001-5168-9846)
-# Version:     2.0.0
-# Last update: 2025-04-15 07:12
+# Version:     2.0.1
+# Last update: 2025-04-15 18:43
 # Copyright (c) 2025 Thomas Wieland
 #------------------------------------------------------------------------------------------
 

{diffindiff-2.0.0 → diffindiff-2.0.1}/diffindiff.egg-info/PKG-INFO

@@ -1,10 +1,24 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: diffindiff
-Version: 2.0.0
+Version: 2.0.1
 Summary: diffindiff: Python library for convenient Difference-in-Differences Analyses
 Author: Thomas Wieland
 Author-email: geowieland@googlemail.com
 Description-Content-Type: text/markdown
+Requires-Dist: numpy
+Requires-Dist: pandas
+Requires-Dist: statsmodels
+Requires-Dist: matplotlib
+Requires-Dist: datetime
+Requires-Dist: scikit-learn
+Requires-Dist: xgboost
+Requires-Dist: lightgbm
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-dist
+Dynamic: summary
 
 # diffindiff: Difference-in-Differences (DiD) Analysis Python Library
 

{diffindiff-2.0.0 → diffindiff-2.0.1}/setup.py

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