Merge pull request #670 from tillahoffmann/lp__

Fix `lp__` test.
stan-dev · Jun 11, 2023 · 7ad7dcc · 7ad7dcc
2 parents cd65084 + 7951466
commit 7ad7dcc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cmdstanpy/model.py b/cmdstanpy/model.py
@@ -1625,7 +1625,7 @@ def log_prob(
             either as a dictionary with entries matching the data variables,
             or as the path of a data file in JSON or Rdump format.
 
-        :return: A pandas.DataFrame containing columns "lp_" and additional
+        :return: A pandas.DataFrame containing columns "lp__" and additional
             columns for the gradient values. These gradients will be for the
             unconstrained parameters of the model.
         """

diff --git a/test/test_log_prob.py b/test/test_log_prob.py
@@ -22,7 +22,7 @@
 def test_lp_good() -> None:
     model = CmdStanModel(stan_file=BERN_STAN)
     x = model.log_prob({"theta": 0.1}, data=BERN_DATA)
-    assert "lp_" in x.columns
+    assert "lp__" in x.columns
 
 
 def test_lp_bad(