Skip to content

Commit

Permalink
remove max_treedepth
Browse files Browse the repository at this point in the history
  • Loading branch information
nitishp25 authored and OriolAbril committed Oct 2, 2020
1 parent 676b18c commit 2491ceb
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions arviz/data/io_pystan.py
Original file line number Diff line number Diff line change
Expand Up @@ -639,7 +639,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:
Expand Down Expand Up @@ -772,7 +771,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

Expand Down

0 comments on commit 2491ceb

Please sign in to comment.