Skip to content

Commit 4540b84

Browse files
committed
renamed samples argument name and pathfinder variables to avoid confusion
1 parent 5aa9544 commit 4540b84

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pymc_experimental/inference/pathfinder.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def convert_flat_trace_to_idata(
6262

6363

6464
def fit_pathfinder(
65-
samples=1000,
65+
num_samples=1000,
6666
random_seed: RandomSeed | None = None,
6767
postprocessing_backend="cpu",
6868
model=None,
@@ -120,14 +120,14 @@ def logprob_fn(x):
120120
initial_position=ip_map.data,
121121
**pathfinder_kwargs,
122122
)
123-
samples, _ = blackjax.vi.pathfinder.sample(
123+
pathfinder_samples, _ = blackjax.vi.pathfinder.sample(
124124
rng_key=jax.random.key(sample_seed),
125125
state=pathfinder_state,
126-
num_samples=samples,
126+
num_samples=num_samples,
127127
)
128128

129129
idata = convert_flat_trace_to_idata(
130-
samples,
130+
pathfinder_samples,
131131
postprocessing_backend=postprocessing_backend,
132132
model=model,
133133
)

0 commit comments

Comments
 (0)