Skip to content

Commit 8ae8cc4

Browse files
committed
Change logic to match cmdstanr discussion
1 parent 39dd54b commit 8ae8cc4

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

cmdstanpy/model.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1637,7 +1637,7 @@ def pathfinder(
16371637
:param draws: Number of approximate draws to return.
16381638
16391639
:param num_single_draws: Number of draws each single-pathfinder will
1640-
draw. By default, this is set to be equal to draws.
1640+
draw.
16411641
If ``num_paths`` is 1, only one of this and ``draws`` should be
16421642
used.
16431643
@@ -1732,13 +1732,14 @@ def pathfinder(
17321732
"before 2.33"
17331733
)
17341734

1735-
if num_single_draws is None:
1736-
num_single_draws = draws
1737-
elif num_paths == 1 and draws is not None and num_single_draws != draws:
1738-
raise ValueError(
1739-
"Cannot specify both 'draws' and 'num_single_draws'"
1740-
" when 'num_paths' is 1"
1741-
)
1735+
if num_paths == 1:
1736+
if num_single_draws is None:
1737+
num_single_draws = draws
1738+
if draws is not None and num_single_draws != draws:
1739+
raise ValueError(
1740+
"Cannot specify both 'draws' and 'num_single_draws'"
1741+
" when 'num_paths' is 1"
1742+
)
17421743

17431744
pathfinder_args = PathfinderArgs(
17441745
init_alpha=init_alpha,

0 commit comments

Comments
 (0)