Skip to content

Commit 367bdd6

Browse files
ferrinetwiecki
authored andcommitted
fix the docstring and args 1.
1 parent c120b02 commit 367bdd6

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

pymc/variational/inference.py

-9
Original file line numberDiff line numberDiff line change
@@ -668,7 +668,6 @@ def run_profiling(self, n=1000, score=None, obj_n_mc=500, **kwargs):
668668

669669
def fit(
670670
n=10000,
671-
local_rv=None,
672671
method="advi",
673672
model=None,
674673
random_seed=None,
@@ -682,19 +681,13 @@ def fit(
682681
----------
683682
n: `int`
684683
number of iterations
685-
local_rv: dict[var->tuple]
686-
mapping {model_variable -> approx params}
687-
Local Vars are used for Autoencoding Variational Bayes
688-
See (AEVB; Kingma and Welling, 2014) for details
689684
method: str or :class:`Inference`
690685
string name is case insensitive in:
691686
692687
- 'advi' for ADVI
693688
- 'fullrank_advi' for FullRankADVI
694689
- 'svgd' for Stein Variational Gradient Descent
695690
- 'asvgd' for Amortized Stein Variational Gradient Descent
696-
- 'nfvi' for Normalizing Flow with default `scale-loc` flow
697-
- 'nfvi=<formula>' for Normalizing Flow using formula
698691
699692
model: :class:`Model`
700693
PyMC model for inference
@@ -743,8 +736,6 @@ def fit(
743736
inf_kwargs = dict()
744737
else:
745738
inf_kwargs = inf_kwargs.copy()
746-
if local_rv is not None:
747-
inf_kwargs["local_rv"] = local_rv
748739
if random_seed is not None:
749740
inf_kwargs["random_seed"] = random_seed
750741
if start is not None:

0 commit comments

Comments
 (0)