From a19d7644158ad1af130f65a0dcbf66751a9c8e29 Mon Sep 17 00:00:00 2001
From: GStechschulte <gabriel.stechschulte@hslu.ch>
Date: Tue, 7 Jan 2025 10:49:54 -0500
Subject: [PATCH] rename package name

---
 .github/workflows/CI.yml                           | 8 ++++----
 benchmark/examples/bart_examples.py                | 4 ++--
 pyproject.toml                                     | 2 +-
 python/{pymc_bart => pymc_bart_rs}/__init__.py     | 8 ++++----
 python/{pymc_bart => pymc_bart_rs}/bart.py         | 0
 python/{pymc_bart => pymc_bart_rs}/compile_pymc.py | 0
 python/{pymc_bart => pymc_bart_rs}/pgbart.py       | 6 +++---
 python/{pymc_bart => pymc_bart_rs}/utils.py        | 0
 8 files changed, 14 insertions(+), 14 deletions(-)
 rename python/{pymc_bart => pymc_bart_rs}/__init__.py (76%)
 rename python/{pymc_bart => pymc_bart_rs}/bart.py (100%)
 rename python/{pymc_bart => pymc_bart_rs}/compile_pymc.py (100%)
 rename python/{pymc_bart => pymc_bart_rs}/pgbart.py (97%)
 rename python/{pymc_bart => pymc_bart_rs}/utils.py (100%)

diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index b95ef6f..eee014d 100644
--- a/.github/workflows/CI.yml
+++ b/.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
 
diff --git a/benchmark/examples/bart_examples.py b/benchmark/examples/bart_examples.py
index 542984b..182cd19 100644
--- a/benchmark/examples/bart_examples.py
+++ b/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)
 
diff --git a/pyproject.toml b/pyproject.toml
index 37d6dea..950f4d2 100644
--- a/pyproject.toml
+++ b/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",
diff --git a/python/pymc_bart/__init__.py b/python/pymc_bart_rs/__init__.py
similarity index 76%
rename from python/pymc_bart/__init__.py
rename to python/pymc_bart_rs/__init__.py
index 190b6bc..e7fe98f 100644
--- a/python/pymc_bart/__init__.py
+++ b/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,
diff --git a/python/pymc_bart/bart.py b/python/pymc_bart_rs/bart.py
similarity index 100%
rename from python/pymc_bart/bart.py
rename to python/pymc_bart_rs/bart.py
diff --git a/python/pymc_bart/compile_pymc.py b/python/pymc_bart_rs/compile_pymc.py
similarity index 100%
rename from python/pymc_bart/compile_pymc.py
rename to python/pymc_bart_rs/compile_pymc.py
diff --git a/python/pymc_bart/pgbart.py b/python/pymc_bart_rs/pgbart.py
similarity index 97%
rename from python/pymc_bart/pgbart.py
rename to python/pymc_bart_rs/pgbart.py
index 18fa94d..c6e6bff 100644
--- a/python/pymc_bart/pgbart.py
+++ b/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):
diff --git a/python/pymc_bart/utils.py b/python/pymc_bart_rs/utils.py
similarity index 100%
rename from python/pymc_bart/utils.py
rename to python/pymc_bart_rs/utils.py