remove max_treedepth

arviz-devs · Apr 27, 2020 · 000776a · 000776a
1 parent a32889b
commit 000776a
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/arviz/data/io_pymc3.py b/arviz/data/io_pymc3.py
@@ -201,7 +201,6 @@ def sample_stats_to_xarray(self):
         for stat in self.trace.stat_names:
             name = rename_key.get(stat, stat)
             data[name] = np.array(self.trace.get_sampler_stats(stat, combine=False))
-        data.update({"max_treedepth": False})
 
         return dict_to_dataset(data, library=self.pymc3, dims=None, coords=self.coords)
 

diff --git a/arviz/data/io_pystan.py b/arviz/data/io_pystan.py
@@ -609,7 +609,6 @@ def get_sample_stats(fit, warmup=False):
         name = re.sub("__$", "", key)
         name = "diverging" if name == "divergent" else name
         data[name] = values
-    data.update({"max_treedepth": False})
 
     data_warmup = OrderedDict()
     if warmup:
@@ -679,7 +678,6 @@ def get_sample_stats_stan3(fit, variables=None, ignore=None):
         name = re.sub("__$", "", key)
         name = "diverging" if name == "divergent" else name
         data[name] = values
-    data.update({"max_treedepth": False})
 
     return data