Find step_size if zero (#348)

* Find step_size if zero

* Update abstractmcmc.jl

* Apply suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

* Update Project.toml

* Update abstractmcmc.jl

* Apply suggestions from code review

Co-authored-by: Tor Erlend Fjelde <[email protected]>

* reuse some code.

---------

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: Tor Erlend Fjelde <[email protected]>

Author: 3 people

Project.toml

@@ -1,6 +1,6 @@
 name = "AdvancedHMC"
 uuid = "0bf59076-c3b1-5ca4-86bd-e02cd72cde3d"
-version = "0.5.4"
+version = "0.5.5"
 
 [deps]
 AbstractMCMC = "80f14c24-f653-4e6a-9b94-39d6b0f70001"

src/abstractmcmc.jl

@@ -273,7 +273,9 @@ function make_step_size(
     hamiltonian::Hamiltonian,
     init_params,
 )
-    return spl.κ.τ.integrator.ϵ
+    T = typeof(spl.κ.τ.integrator.ϵ)
+    ϵ = make_step_size(rng, spl.κ.τ.integrator, T, hamiltonian, init_params)
+    return ϵ
 end
 
 function make_step_size(
@@ -294,7 +296,13 @@ function make_step_size(
     hamiltonian::Hamiltonian,
     init_params,
 )
-    return integrator.ϵ
+    if integrator.ϵ > 0
+        ϵ = integrator.ϵ
+    else
+        ϵ = find_good_stepsize(rng, hamiltonian, init_params)
+        @info string("Found initial step size ", ϵ)
+    end
+    return T(ϵ)
 end
 
 function make_step_size(