From 63136d4be2c32c6662c1c475533a3a3a81d1c731 Mon Sep 17 00:00:00 2001 From: Drew Herren Date: Thu, 5 Dec 2024 17:01:27 -0600 Subject: [PATCH] Fixed bug in python BART serialization --- stochtree/bart.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stochtree/bart.py b/stochtree/bart.py index 6d78dae8..e4fbaec2 100644 --- a/stochtree/bart.py +++ b/stochtree/bart.py @@ -706,7 +706,7 @@ def to_json(self) -> str: # Add parameter samples if self.sample_sigma_global: bart_json.add_numeric_vector("sigma2_global_samples", self.global_var_samples, "parameters") - if self.sample_sigma_global: + if self.sample_sigma_leaf: bart_json.add_numeric_vector("sigma2_leaf_samples", self.leaf_scale_samples, "parameters") return bart_json.return_json_string() @@ -758,7 +758,7 @@ def from_json(self, json_string: str) -> None: # Unpack parameter samples if self.sample_sigma_global: self.global_var_samples = bart_json.get_numeric_vector("sigma2_global_samples", "parameters") - if self.sample_sigma_global: + if self.sample_sigma_leaf: self.leaf_scale_samples = bart_json.get_numeric_vector("sigma2_leaf_samples", "parameters") # Mark the deserialized model as "sampled"