Merge pull request #156 from StochasticTree/python-update-0.1.1

Preparing for 0.1.1 Release

Author: andrewherren

demo/notebooks/causal_inference.ipynb

@@ -106,13 +106,13 @@
     "bcf_model = BCFModel()\n",
     "general_params = {\"keep_every\": 5}\n",
     "bcf_model.sample(\n",
-    "    X_train,\n",
-    "    Z_train,\n",
-    "    y_train,\n",
-    "    pi_train,\n",
-    "    X_test,\n",
-    "    Z_test,\n",
-    "    pi_test,\n",
+    "    X_train=X_train,\n",
+    "    Z_train=Z_train,\n",
+    "    y_train=y_train,\n",
+    "    pi_train=pi_train,\n",
+    "    X_test=X_test,\n",
+    "    Z_test=Z_test,\n",
+    "    pi_test=pi_test,\n",
     "    num_gfr=10,\n",
     "    num_mcmc=100,\n",
     "    general_params=general_params,\n",
@@ -236,7 +236,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.17"
+   "version": "3.12.9"
   }
  },
  "nbformat": 4,

demo/notebooks/causal_inference_feature_subsets.ipynb

@@ -110,13 +110,13 @@
    "source": [
     "bcf_model = BCFModel()\n",
     "bcf_model.sample(\n",
-    "    X_train,\n",
-    "    Z_train,\n",
-    "    y_train,\n",
-    "    pi_train,\n",
-    "    X_test,\n",
-    "    Z_test,\n",
-    "    pi_test,\n",
+    "    X_train=X_train,\n",
+    "    Z_train=Z_train,\n",
+    "    y_train=y_train,\n",
+    "    pi_train=pi_train,\n",
+    "    X_test=X_test,\n",
+    "    Z_test=Z_test,\n",
+    "    pi_test=pi_test,\n",
     "    num_gfr=10,\n",
     "    num_mcmc=100,\n",
     "    general_params={\"keep_every\": 5},\n",
@@ -239,13 +239,13 @@
     "bcf_model_subset = BCFModel()\n",
     "tau_params = {\"keep_vars\": [0, 1]}\n",
     "bcf_model_subset.sample(\n",
-    "    X_train,\n",
-    "    Z_train,\n",
-    "    y_train,\n",
-    "    pi_train,\n",
-    "    X_test,\n",
-    "    Z_test,\n",
-    "    pi_test,\n",
+    "    X_train=X_train,\n",
+    "    Z_train=Z_train,\n",
+    "    y_train=y_train,\n",
+    "    pi_train=pi_train,\n",
+    "    X_test=X_test,\n",
+    "    Z_test=Z_test,\n",
+    "    pi_test=pi_test,\n",
     "    num_gfr=10,\n",
     "    num_mcmc=100,\n",
     "    treatment_effect_forest_params=tau_params,\n",
@@ -369,7 +369,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.17"
+   "version": "3.12.9"
   }
  },
  "nbformat": 4,

demo/notebooks/heteroskedastic_supervised_learning.ipynb

@@ -250,7 +250,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.17"
+   "version": "3.12.9"
   }
  },
  "nbformat": 4,

demo/notebooks/multivariate_treatment_causal_inference.ipynb

@@ -107,13 +107,13 @@
    "source": [
     "bcf_model = BCFModel()\n",
     "bcf_model.sample(\n",
-    "    X_train,\n",
-    "    Z_train,\n",
-    "    y_train,\n",
-    "    pi_train,\n",
-    "    X_test,\n",
-    "    Z_test,\n",
-    "    pi_test,\n",
+    "    X_train=X_train,\n",
+    "    Z_train=Z_train,\n",
+    "    y_train=y_train,\n",
+    "    pi_train=pi_train,\n",
+    "    X_test=X_test,\n",
+    "    Z_test=Z_test,\n",
+    "    pi_test=pi_test,\n",
     "    num_gfr=10,\n",
     "    num_mcmc=100,\n",
     ")"
@@ -260,7 +260,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "stochtree-dev",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },