Correct the order of break (#157)

* fix order of break

* increment version

Author: xukai92

Project.toml

@@ -1,6 +1,6 @@
 name = "AdvancedHMC"
 uuid = "0bf59076-c3b1-5ca4-86bd-e02cd72cde3d"
-version = "0.2.17"
+version = "0.2.18"
 
 [deps]
 ArgCheck = "dce04be8-c92d-5529-be00-80e4d2c0e197"

src/integrator.jl

@@ -66,12 +66,13 @@ function step(
         r = temper(lf, r, (i=i, is_half=false), n_steps)
         # Create a new phase point by caching the logdensity and gradient
         z = phasepoint(h, θ, r; ℓπ=DualValue(value, gradient))
-        !isfinite(z) && break
+        # Update result
         if res isa Vector
             res[i] = z
         else
             res = z
         end
+        !isfinite(z) && break
     end
     res
 end