csdid 0.2.7__tar.gz → 0.2.9__tar.gz

{csdid-0.2.7 → csdid-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csdid
-Version: 0.2.7
+Version: 0.2.9
 Summary: Difference in Difference in Python
 Home-page: https://github.com/d2cml-ai/csdid
 Author: D2CML Team, Alexander Quispe, Carlos Guevara, Jhon Flroes
@@ -13,7 +13,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Topic :: Scientific/Engineering
 License-File: LICENSE
 Requires-Dist: pandas
-Requires-Dist: numpy<=1.24.3
+Requires-Dist: numpy
 Requires-Dist: scipy
 Requires-Dist: patsy
 Requires-Dist: plotnine

csdid-0.2.9/csdid/_version.py

@@ -0,0 +1 @@
+__version = '0.2.9'

{csdid-0.2.7 → csdid-0.2.9}/csdid/att_gt.py

@@ -32,7 +32,6 @@ class ATTgt:
     dp['alp'] = alp
     dp['true_repeated_cross_sections'] = dp['true_rep_cross_section']
     dp['cband'] = cband
-    dp['panel'] = panel
     self.dp = dp
 
   def fit(self, est_method = 'dr', base_period = 'varying', bstrap = True):

{csdid-0.2.7 → csdid-0.2.9}/csdid/attgt_fnc/preprocess_did.py

@@ -261,7 +261,6 @@ def pre_process_did(yname, tname, idname, gname, data: pd.DataFrame,
   n, t = data.shape
   control_group = control_group[0]
   columns = [idname, tname, yname, gname]
-  control_group = "nevertreated"
   # print(columns)
   # Columns
   if clustervar is not None:
@@ -426,4 +425,4 @@ def pre_process_did(yname, tname, idname, gname, data: pd.DataFrame,
     'true_rep_cross_section': true_rep_cross_section,
     'clustervars': clustervar
   }
-  return did_params
+  return did_params

{csdid-0.2.7 → csdid-0.2.9}/csdid/plots/gplot.py

@@ -15,27 +15,42 @@ def gplot(ssresults, ax, ylim=None, xlab=None, ylab=None, title="Group", xgap=1,
     if ylab is None:
         ylab = 'ATT'
     
+    # drop any null years, then overwrite 'year' cleanly as integer
     ssresults = ssresults[ssresults['year'].notnull()].copy()
-    ssresults.loc[:, 'year'] = ssresults['year'].astype(int).astype(str)
+    ssresults = ssresults.assign(year=ssresults['year'].astype(int))
     
-    pre_points = ssresults.loc[ssresults['post'] == 0]
-    post_points = ssresults.loc[ssresults['post'] == 1]
-    
-    ax.errorbar(pre_points['year'], pre_points['att'], yerr=pre_points['c']*pre_points['att_se'],
-                 fmt='o', markersize=5, color='#e87d72', ecolor='#e87d72', capsize=5, label='Pre')   
+    pre_points  = ssresults[ssresults['post'] == 0]
+    post_points = ssresults[ssresults['post'] == 1]
     
-    ax.errorbar(post_points['year'], post_points['att'], yerr=post_points['c']*post_points['att_se'],
-                 fmt='o', markersize=5, color='#56bcc2', ecolor='#56bcc2', capsize=5, label='Post')  
+    # plot Pre
+    ax.errorbar(
+        pre_points['year'].to_numpy(),
+        pre_points['att'].to_numpy(),
+        yerr=(pre_points['c'] * pre_points['att_se']).to_numpy(),
+        fmt='o', markersize=5,
+        color='#e87d72', ecolor='#e87d72',
+        capsize=5, label='Pre'
+    )
+    # plot Post
+    ax.errorbar(
+        post_points['year'].to_numpy(),
+        post_points['att'].to_numpy(),
+        yerr=(post_points['c'] * post_points['att_se']).to_numpy(),
+        fmt='o', markersize=5,
+        color='#56bcc2', ecolor='#56bcc2',
+        capsize=5, label='Post'
+    )
     
     ax.set_ylim(ylim)
     ax.set_title(title)
-    ax.set_xlabel(xlab)    
-    ax.set_ylabel(ylab)    
+    ax.set_xlabel(xlab)
+    ax.set_ylabel(ylab)
 
-    handles, labels = ax.get_legend_handles_labels()    
+    handles, labels = ax.get_legend_handles_labels()
     
     if ref_line is not None:
         ax.axhline(ref_line, linestyle='dashed', color='#1F1F1F')
+
     if theming:
         ax.set_facecolor('white')
         ax.set_title(title, color="#1F1F1F", fontweight="bold", fontsize=10)
@@ -43,18 +58,26 @@ def gplot(ssresults, ax, ylim=None, xlab=None, ylab=None, title="Group", xgap=1,
         ax.spines['left'].set_color('#1F1F1F')
         ax.tick_params(axis='x', colors='#1F1F1F')
         ax.tick_params(axis='y', colors='#1F1F1F')
+
+        # draw legend below if both series exist
         if not pre_points.empty and not post_points.empty:
-            ax.legend(handles[0:2], labels[0:2], loc='lower center',fontsize='small', ncol=2, bbox_to_anchor=(0.5,-0.27))
+            ax.legend(handles[0:2], labels[0:2],
+                      loc='lower center', fontsize='small',
+                      ncol=2, bbox_to_anchor=(0.5, -0.27))
         elif not pre_points.empty:
-            ax.legend(handles[:1], labels[:1], loc='lower center',fontsize='small', ncol=2, bbox_to_anchor=(0.5,-0.27))
+            ax.legend(handles[:1], labels[:1],
+                      loc='lower center', fontsize='small',
+                      ncol=2, bbox_to_anchor=(0.5, -0.27))
         elif not post_points.empty:
-            ax.legend(handles[1:2], labels[1:2], loc='lower center',fontsize='small', ncol=2, bbox_to_anchor=(0.5,-0.27))     
+            ax.legend(handles[1:2], labels[1:2],
+                      loc='lower center', fontsize='small',
+                      ncol=2, bbox_to_anchor=(0.5, -0.27))
+
     if not legend:
         ax.legend().set_visible(False)
         
     return ax
 
-
 def splot(ssresults, ax, ylim=None, xlab=None, ylab=None, title="Group",
           legend=True, ref_line=0, theming=True):
     

{csdid-0.2.7 → csdid-0.2.9}/csdid.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csdid
-Version: 0.2.7
+Version: 0.2.9
 Summary: Difference in Difference in Python
 Home-page: https://github.com/d2cml-ai/csdid
 Author: D2CML Team, Alexander Quispe, Carlos Guevara, Jhon Flroes
@@ -13,7 +13,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Topic :: Scientific/Engineering
 License-File: LICENSE
 Requires-Dist: pandas
-Requires-Dist: numpy<=1.24.3
+Requires-Dist: numpy
 Requires-Dist: scipy
 Requires-Dist: patsy
 Requires-Dist: plotnine

{csdid-0.2.7 → csdid-0.2.9}/csdid.egg-info/requires.txt

@@ -1,5 +1,5 @@
 pandas
-numpy<=1.24.3
+numpy
 scipy
 patsy
 plotnine

{csdid-0.2.7 → csdid-0.2.9}/setup.py

@@ -23,13 +23,12 @@ setup(
     ],
   install_requires=[
         'pandas',
-        'numpy<=1.24.3',
+        'numpy',
         'scipy',
         'patsy',
         'plotnine',
         'twine',
-        'joblib',
-        #'drdid @ git+https://github.com/d2cml-ai/DRDID'
+        'joblib'
   ],
   packages=find_packages(),
   package_data={

csdid-0.2.7/csdid/_version.py

@@ -1 +0,0 @@
-__version = '0.2.7'