split rules error handling

GStechschulte · Jan 7, 2025 · 17d5b4f · 17d5b4f
1 parent 2b51547
commit 17d5b4f
Show file tree

Hide file tree

Showing 89 changed files with 95 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -14,7 +14,7 @@
 
 [package]
 name = "pymc-bart"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 
 [lib]

diff --git a/benchmark/environments/bart_rust.yml b/benchmark/environments/bart_rust.yml
@@ -258,7 +258,7 @@ dependencies:
   - pip:
       - py-spy==0.4.0
       - pymc==5.19.1
-      - pymc-bart==0.1.0
+      - pymc-bart-rs==0.1.1
       - pytensor==2.26.4
       - ruff==0.8.3
       - split-rules==0.1.0
diff --git a/benchmark/examples/bart_examples.py b/benchmark/examples/bart_examples.py
@@ -73,7 +73,13 @@ def test_coal(args):
     y_data = hist
 
     with pm.Model() as model_coal:
-        mu = pmb.BART("mu", X=x_data, Y=np.log(y_data), m=args.trees, split_rules=["ContinuousSplit"])
+        mu = pmb.BART(
+            "mu",
+            X=x_data,
+            Y=np.log(y_data),
+            m=args.trees,
+            # split_rules=["ContinuousSplit"]
+        )
         exp_mu = pm.Deterministic("exp_mu", pm.math.exp(mu))
         y_pred = pm.Poisson("y_pred", mu=exp_mu, observed=y_data)
 

diff --git a/...nchmark_20250107_172853/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...nchmark_20250107_172853/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+2
diff --git a/...enchmark_20250107_172853/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...enchmark_20250107_172853/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+35
diff --git a/...rk_20250107_172853/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...rk_20250107_172853/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+3
diff --git a/...chmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.1_0.1_model_bikes.txt b/...chmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.1_0.1_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.1_0.1_model_coal.txt b/...nchmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.1_0.1_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.1_0.1_model_propensity.txt b/...k_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.1_0.1_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.5_0.5_model_bikes.txt b/...chmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.5_0.5_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.5_0.5_model_coal.txt b/...nchmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.5_0.5_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.5_0.5_model_propensity.txt b/...k_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch0.5_0.5_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...chmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...nchmark_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...k_20250107_173328/benchmark_t100_p20_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.1_0.1_model_bikes.txt b/...chmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.1_0.1_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.1_0.1_model_coal.txt b/...nchmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.1_0.1_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.1_0.1_model_propensity.txt b/...k_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.1_0.1_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.5_0.5_model_bikes.txt b/...chmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.5_0.5_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.5_0.5_model_coal.txt b/...nchmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.5_0.5_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.5_0.5_model_propensity.txt b/...k_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch0.5_0.5_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...chmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...nchmark_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...k_20250107_173328/benchmark_t100_p40_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.1_0.1_model_bikes.txt b/...chmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.1_0.1_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.1_0.1_model_coal.txt b/...nchmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.1_0.1_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.1_0.1_model_propensity.txt b/...k_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.1_0.1_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.5_0.5_model_bikes.txt b/...chmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.5_0.5_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.5_0.5_model_coal.txt b/...nchmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.5_0.5_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.5_0.5_model_propensity.txt b/...k_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch0.5_0.5_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...chmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...nchmark_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...k_20250107_173328/benchmark_t100_p60_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.1_0.1_model_bikes.txt b/...chmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.1_0.1_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.1_0.1_model_coal.txt b/...nchmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.1_0.1_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.1_0.1_model_propensity.txt b/...k_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.1_0.1_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.5_0.5_model_bikes.txt b/...chmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.5_0.5_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.5_0.5_model_coal.txt b/...nchmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.5_0.5_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.5_0.5_model_propensity.txt b/...k_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch0.5_0.5_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...chmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...nchmark_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...k_20250107_173328/benchmark_t200_p20_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.1_0.1_model_bikes.txt b/...chmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.1_0.1_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.1_0.1_model_coal.txt b/...nchmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.1_0.1_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.1_0.1_model_propensity.txt b/...k_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.1_0.1_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.5_0.5_model_bikes.txt b/...chmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.5_0.5_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.5_0.5_model_coal.txt b/...nchmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.5_0.5_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.5_0.5_model_propensity.txt b/...k_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch0.5_0.5_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...chmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...nchmark_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...k_20250107_173328/benchmark_t200_p40_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.1_0.1_model_bikes.txt b/...chmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.1_0.1_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.1_0.1_model_coal.txt b/...nchmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.1_0.1_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.1_0.1_model_propensity.txt b/...k_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.1_0.1_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.5_0.5_model_bikes.txt b/...chmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.5_0.5_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.5_0.5_model_coal.txt b/...nchmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.5_0.5_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.5_0.5_model_propensity.txt b/...k_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch0.5_0.5_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...chmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...chmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...nchmark_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...k_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...k_20250107_173328/benchmark_t200_p60_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.1_0.1_model_bikes.txt b/...nchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.1_0.1_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...enchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.1_0.1_model_coal.txt b/...enchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.1_0.1_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...rk_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.1_0.1_model_propensity.txt b/...rk_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.1_0.1_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.5_0.5_model_bikes.txt b/...nchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.5_0.5_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...enchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.5_0.5_model_coal.txt b/...enchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.5_0.5_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...rk_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.5_0.5_model_propensity.txt b/...rk_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch0.5_0.5_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...nchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...enchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...enchmark_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...rk_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...rk_20250107_173328/benchmark_t50_p20_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.1_0.1_model_bikes.txt b/...nchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.1_0.1_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...enchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.1_0.1_model_coal.txt b/...enchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.1_0.1_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...rk_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.1_0.1_model_propensity.txt b/...rk_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.1_0.1_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.5_0.5_model_bikes.txt b/...nchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.5_0.5_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...enchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.5_0.5_model_coal.txt b/...enchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.5_0.5_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...rk_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.5_0.5_model_propensity.txt b/...rk_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch0.5_0.5_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...nchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...enchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...enchmark_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...rk_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...rk_20250107_173328/benchmark_t50_p40_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.1_0.1_model_bikes.txt b/...nchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.1_0.1_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...enchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.1_0.1_model_coal.txt b/...enchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.1_0.1_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...rk_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.1_0.1_model_propensity.txt b/...rk_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.1_0.1_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.5_0.5_model_bikes.txt b/...nchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.5_0.5_model_bikes.txt
@@ -0,0 +1 @@
+1
diff --git a/...enchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.5_0.5_model_coal.txt b/...enchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.5_0.5_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...rk_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.5_0.5_model_propensity.txt b/...rk_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch0.5_0.5_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/...nchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch1.0_1.0_model_bikes.txt b/...nchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch1.0_1.0_model_bikes.txt
@@ -0,0 +1 @@
+0
diff --git a/...enchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch1.0_1.0_model_coal.txt b/...enchmark_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch1.0_1.0_model_coal.txt
@@ -0,0 +1 @@
+0
diff --git a/...rk_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch1.0_1.0_model_propensity.txt b/...rk_20250107_173328/benchmark_t50_p60_tune1000_draws1000_batch1.0_1.0_model_propensity.txt
@@ -0,0 +1 @@
+0
diff --git a/python/pymc_bart/bart.py b/python/pymc_bart/bart.py
@@ -118,8 +118,8 @@ def __new__(
         alpha: float = 0.95,
         beta: float = 2.0,
         response: str = "constant",
+        split_rules: List[str] = ["ContinuousSplit"],
         split_prior: Optional[npt.NDArray[np.float_]] = None,
-        split_rules: Optional[List] = None,
         separate_trees: Optional[bool] = False,
         **kwargs,
     ):