File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
pymc_experimental/inference Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -62,7 +62,7 @@ def convert_flat_trace_to_idata(
62
62
63
63
64
64
def fit_pathfinder (
65
- samples = 1000 ,
65
+ num_samples = 1000 ,
66
66
random_seed : RandomSeed | None = None ,
67
67
postprocessing_backend = "cpu" ,
68
68
model = None ,
@@ -120,14 +120,14 @@ def logprob_fn(x):
120
120
initial_position = ip_map .data ,
121
121
** pathfinder_kwargs ,
122
122
)
123
- samples , _ = blackjax .vi .pathfinder .sample (
123
+ pathfinder_samples , _ = blackjax .vi .pathfinder .sample (
124
124
rng_key = jax .random .key (sample_seed ),
125
125
state = pathfinder_state ,
126
- num_samples = samples ,
126
+ num_samples = num_samples ,
127
127
)
128
128
129
129
idata = convert_flat_trace_to_idata (
130
- samples ,
130
+ pathfinder_samples ,
131
131
postprocessing_backend = postprocessing_backend ,
132
132
model = model ,
133
133
)
You can’t perform that action at this time.
0 commit comments