update doc hmc

pyro-ppl · Mar 1, 2024 · 5c34a54 · 5c34a54
1 parent 44249e9
commit 5c34a54
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyro/infer/mcmc/hmc.py b/pyro/infer/mcmc/hmc.py
@@ -66,8 +66,8 @@ class HMC(MCMCKernel):
         step size, hence the sampling will be slower and more robust. Default to 0.8.
     :param callable init_strategy: A per-site initialization function.
         See :ref:`autoguide-initialization` section for available functions.
-    :param min_stepsize (float): Lower bound on stepsize in adaptation strategy.
-    :param max_stepsize (float): Upper bound on stepsize in adaptation strategy.
+    :param float min_stepsize: Lower bound on stepsize in adaptation strategy.
+    :param float max_stepsize: Upper bound on stepsize in adaptation strategy.
 
     .. note:: Internally, the mass matrix will be ordered according to the order
         of the names of latent variables, not the order of their appearance in