Skip to content

Commit 03ad8dd

Browse files
michaelraczyckitwiecki
authored andcommitted
removed unused *args from fit, added named args to sampler_config
1 parent b3be15f commit 03ad8dd

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pymc_experimental/model_builder.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,6 @@ def fit(
236236
progressbar: bool = True,
237237
random_seed: RandomState = None,
238238
data: Dict[str, Union[np.ndarray, pd.DataFrame, pd.Series]] = None,
239-
*args: Any,
240239
**kwargs: Any,
241240
) -> az.InferenceData:
242241
"""
@@ -270,6 +269,10 @@ def fit(
270269
if self.sampler_config is None:
271270
self.sampler_config = sampler_config
272271
self.build_model(self.model_data, self.model_config)
272+
273+
sampler_config["progressbar"] = progressbar
274+
sampler_config["random_seed"] = random_seed
275+
273276
with self.model:
274277
self.idata = pm.sample(**self.sampler_config, **kwargs)
275278
self.idata.extend(pm.sample_prior_predictive())

0 commit comments

Comments
 (0)