rename package name

.github/workflows/CI.yml

@@ -41,7 +41,7 @@ jobs:
         shell: bash
         run: |
           set -e
-          pip install --find-links dist --force-reinstall 'pymc-bart[all]'
+          pip install --find-links dist --force-reinstall 'pymc-bart-rs[all]'
           pip install pytest
           pytest
       - name: pytest
@@ -57,7 +57,7 @@ jobs:
             pip3 install -U pip pytest
           run: |
             set -e
-            pip3 install --find-links dist --force-reinstall 'pymc-bart[all]'
+            pip3 install --find-links dist --force-reinstall 'pymc-bart-rs[all]'
             pytest
 
   windows:
@@ -88,7 +88,7 @@ jobs:
         shell: bash
         run: |
           set -e
-          pip install "pymc-bart[all]" --find-links dist --force-reinstall
+          pip install "pymc-bart-rs[all]" --find-links dist --force-reinstall
           pip install pytest
           pytest
 
@@ -119,7 +119,7 @@ jobs:
         shell: bash
         run: |
           set -e
-          pip install 'pymc-bart[all]' --find-links dist --force-reinstall
+          pip install 'pymc-bart-rs[all]' --find-links dist --force-reinstall
           pip install pytest
           pytest
 

benchmark/examples/bart_examples.py

@@ -5,7 +5,7 @@
 import numpy as np
 import pandas as pd
 import pymc as pm
-import pymc_bart as pmb
+import pymc_bart_rs as pmb
 
 
 RANDOM_SEED = 8457
@@ -74,7 +74,7 @@ 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)
+        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)
 

pyproject.toml

@@ -3,7 +3,7 @@ requires = ["maturin>=1.7,<2.0"]
 build-backend = "maturin"
 
 [project]
-name = "pymc_bart"
+name = "pymc_bart_rs"
 requires-python = ">=3.8, <3.13"
 classifiers = [
     "Programming Language :: Rust",

python/pymc_bart/__init__.py → python/pymc_bart_rs/__init__.py

@@ -1,9 +1,9 @@
 import pymc as pm
 
-from pymc_bart.bart import BART
-from pymc_bart.compile_pymc import CompiledPyMCModel
-from pymc_bart.pgbart import PGBART
-from pymc_bart.utils import (
+from pymc_bart_rs.bart import BART
+from pymc_bart_rs.compile_pymc import CompiledPyMCModel
+from pymc_bart_rs.pgbart import PGBART
+from pymc_bart_rs.utils import (
     compute_variable_importance,
     plot_convergence,
     plot_dependence,

python/pymc_bart/pgbart.py → python/pymc_bart_rs/pgbart.py

@@ -24,9 +24,9 @@
 from pymc.step_methods.compound import Competence
 from pytensor.graph.basic import Variable
 
-from pymc_bart.bart import BARTRV
-from pymc_bart.compile_pymc import CompiledPyMCModel
-from pymc_bart.pymc_bart import initialize, step
+from pymc_bart_rs.bart import BARTRV
+from pymc_bart_rs.compile_pymc import CompiledPyMCModel
+from pymc_bart_rs.pymc_bart import initialize, step
 
 
 class PGBART(ArrayStepShared):