From 419fefc98ad4515139556ef056a4cfbfcbf3ddcd Mon Sep 17 00:00:00 2001
From: GStechschulte <gabriel.stechschulte@hslu.ch>
Date: Tue, 7 Jan 2025 18:12:52 -0600
Subject: [PATCH] update README

---
 README.md                            | 2 +-
 benchmark/environments/bart_rust.yml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 56f39d6..a6c411f 100644
--- a/README.md
+++ b/README.md
@@ -22,7 +22,7 @@ Get started by using PyMC-BART to set up a BART model
 
 ```python
 import pymc as pm
-import pymc_bart_rs as pmb
+import pymc_bart as pmb
 
 X, y = ... # Your data replaces "..."
 with pm.Model() as model:
diff --git a/benchmark/environments/bart_rust.yml b/benchmark/environments/bart_rust.yml
index d64fe01..f18c63b 100644
--- 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-rs==0.1.1
+      - pymc-bart-rs==0.1.2
       - pytensor==2.26.4
       - ruff==0.8.3
       - split-rules==0.1.0