Merge pull request #339 from StochasticTree/bart-num-threads-hotfix

Fix num_threads pass-through bug in R BART interface

Author: andrewherren

CHANGELOG.md

@@ -9,7 +9,8 @@
 
 ## Bug Fixes
 
-* Fixed probit offset bug R and Python [#337](https://github.com/StochasticTree/stochtree/pull/337)
+* Fixed num_threads pass-through bug in R BART [#339](https://github.com/StochasticTree/stochtree/pull/339)
+* Fixed probit offset bug in R and Python [#337](https://github.com/StochasticTree/stochtree/pull/337)
 * Standardized multi-chain BCF handling of internal propensity models [#334](https://github.com/StochasticTree/stochtree/pull/334)
 * Fixed multi-chain BCF bugs with the parametric intercept term in R and Python [#326](https://github.com/StochasticTree/stochtree/pull/326)
 * Fixed indexing bugs for multivariate treatment BCF in Python [#326](https://github.com/StochasticTree/stochtree/pull/326)

NEWS.md

@@ -7,7 +7,8 @@
 
 ## Bug Fixes
 
-* Fixed probit offset bug R and Python [#337](https://github.com/StochasticTree/stochtree/pull/337)
+* Fixed num_threads pass-through bug in R BART [#339](https://github.com/StochasticTree/stochtree/pull/339)
+* Fixed probit offset bug in R and Python [#337](https://github.com/StochasticTree/stochtree/pull/337)
 * Standardized multi-chain BCF handling of internal propensity models [#334](https://github.com/StochasticTree/stochtree/pull/334)
 * Fixed multi-chain BCF bugs with the parametric intercept term in R and Python [#326](https://github.com/StochasticTree/stochtree/pull/326)
 * Fixed indexing bugs for multivariate treatment BCF in Python [#326](https://github.com/StochasticTree/stochtree/pull/326)

R/bart.R

@@ -1695,6 +1695,7 @@ bart <- function(
           rng = rng,
           forest_model_config = forest_model_config_variance,
           global_model_config = global_model_config,
+          num_threads = num_threads,
           keep_forest = keep_sample,
           gfr = TRUE
         )
@@ -2156,6 +2157,7 @@ bart <- function(
             rng = rng,
             forest_model_config = forest_model_config_mean,
             global_model_config = global_model_config,
+            num_threads = num_threads,
             keep_forest = keep_sample,
             gfr = FALSE
           )
@@ -2222,6 +2224,7 @@ bart <- function(
             rng = rng,
             forest_model_config = forest_model_config_variance,
             global_model_config = global_model_config,
+            num_threads = num_threads,
             keep_forest = keep_sample,
             gfr = FALSE
           )