update

causalmatch/matching/match_core/ate.py

@@ -15,7 +15,7 @@
 import pandas as pd
 import warnings
 import statsmodels.api as sm
-from causalmatch.matching.match_core.utils import data_process_ate
+from .utils import data_process_ate
 warnings.simplefilter(action='ignore', category=FutureWarning)
 
 def ate(match_obj, use_weight, df_post_validate):

causalmatch/matching/match_core/model.py

@@ -23,6 +23,7 @@
 from causalmatch.matching.match_core.utils import data_process_bc, data_process_ate, balance_check_x,gen_test_data
 from causalmatch.matching.match_core.ate import ate
 from causalmatch.matching.match_core.meta_learner import s_learner_linear
+
 import warnings
 
 

notebooks/General Matching Use Case.ipynb

@@ -280,7 +280,7 @@
      "output_type": "stream",
      "text": [
       "[LightGBM] [Info] Number of positive: 4077, number of negative: 3923\n",
-      "[LightGBM] [Info] Auto-choosing col-wise multi-threading, the overhead of testing was 0.000226 seconds.\n",
+      "[LightGBM] [Info] Auto-choosing col-wise multi-threading, the overhead of testing was 0.000239 seconds.\n",
       "You can set `force_col_wise=true` to remove the overhead.\n",
       "[LightGBM] [Info] Total Bins 791\n",
       "[LightGBM] [Info] Number of data points in the train set: 8000, number of used features: 16\n",

setup.py

@@ -10,7 +10,7 @@ def readme():
 
 setup(
     name="causalmatch",
-    version='0.0.1',
+    version='0.0.2',
     author="Xiaoyu Zhou",
     author_email="[email protected]",
     url='https://github.com/bytedance/CausalMatch',