Skip to content

Commit

Permalink
address comment
Browse files Browse the repository at this point in the history
  • Loading branch information
fehiepsi committed Jul 10, 2023
1 parent 3450771 commit 2cf626f
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion numpyro/infer/autoguide.py
Original file line number Diff line number Diff line change
Expand Up @@ -1138,7 +1138,8 @@ def local_model(theta):
:param callable global_guide: A guide for the global latent variables, e.g. an autoguide.
The return type should be a dictionary of latent sample sites names and corresponding samples.
If there is no global variable in the model, we can set this to None.
:param callable local_guide: An optional guide for the local latent variables.
:param callable local_guide: An optional guide for specifying the DAIS base distribution for
local latent variables.
:param str prefix: A prefix that will be prefixed to all internal sites.
:param int K: A positive integer that controls the number of HMC steps used.
Defaults to 4.
Expand Down

0 comments on commit 2cf626f

Please sign in to comment.