Merge pull request #246 from StochasticTree/doc-vignette-hotfix

Update python demos and vignettes

Author: andrewherren

demo/debug/causal_inference.py

@@ -63,7 +63,7 @@
 
 # Run BCF
 bcf_model = BCFModel()
-bcf_model.sample(X_train, Z_train, y_train, pi_train, X_test, Z_test, pi_test, num_gfr=10, num_mcmc=1000)
+bcf_model.sample(X_train = X_train, Z_train = Z_train, y_train = y_train, propensity_train = pi_train, X_test = X_test, Z_test = Z_test, propensity_test = pi_test, num_gfr=10, num_mcmc=1000)
 
 # Inspect the MCMC (BART) samples
 forest_preds_y_mcmc = bcf_model.y_hat_test

demo/debug/multivariate_treatment_causal_inference.py

@@ -43,4 +43,4 @@
 
 # Run BCF
 bcf_model = BCFModel()
-bcf_model.sample(X_train, Z_train, y_train, pi_train, X_test, Z_test, pi_test, num_gfr=10, num_mcmc=100)
+bcf_model.sample(X_train = X_train, Z_train = Z_train, y_train = y_train, propensity_train = pi_train, X_test = X_test, Z_test = Z_test, propensity_test = pi_test, num_gfr=10, num_mcmc=100)

demo/notebooks/multi_chain.ipynb

@@ -488,7 +488,7 @@
     "bcf_model.sample(\n",
     "  X_train = X_train,\n",
     "  Z_train = Z_train,\n",
-    "  pi_train = propensity_train,\n",
+    "  propensity_train = propensity_train,\n",
     "  y_train = y_train,\n",
     "  num_gfr = num_gfr,\n",
     "  num_burnin = num_burnin,\n",
@@ -621,7 +621,7 @@
     "xbcf_model.sample(\n",
     "  X_train = X_train,\n",
     "  Z_train = Z_train,\n",
-    "  pi_train = propensity_train,\n",
+    "  propensity_train = propensity_train,\n",
     "  y_train = y_train,\n",
     "  num_gfr = num_gfr,\n",
     "  num_burnin = 0,\n",
@@ -649,7 +649,7 @@
     "bcf_model.sample(\n",
     "  X_train = X_train,\n",
     "  Z_train = Z_train,\n",
-    "  pi_train = propensity_train,\n",
+    "  propensity_train = propensity_train,\n",
     "  y_train = y_train,\n",
     "  num_gfr = 0,\n",
     "  num_burnin = num_burnin,\n",