Skip to content

Commit

Permalink
Update constructors.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
yebai authored Oct 29, 2023
1 parent 6e5a467 commit 4c03adf
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions test/constructors.jl
Original file line number Diff line number Diff line change
Expand Up @@ -131,9 +131,13 @@ get_kernel_hyperparamsT(spl::NUTS, state) = typeof(state.κ.τ.termination_crite
@test AdvancedHMC.sampler_eltype(sampler) == T

# Step.
transition, state =
AbstractMCMC.step(rng, model, sampler; n_adapts = 0, initial_params = θ_init)

transition, state = AbstractMCMC.step(
rng,
model,
sampler;
n_adapts = 0,
initial_params = θ_init,
)
# Verify that the types are preserved in the transition.
@test eltype(transition.z.θ) == T
@test eltype(transition.z.r) == T
Expand Down

0 comments on commit 4c03adf

Please sign in to comment.