diff --git a/src/AdvancedHMC.jl b/src/AdvancedHMC.jl index b5a25af4..8187c75a 100644 --- a/src/AdvancedHMC.jl +++ b/src/AdvancedHMC.jl @@ -127,32 +127,33 @@ include("constructors.jl") export HMCSampler, HMC, NUTS, HMCDA module Experimental - using Random, Statistics, LinearAlgebra - using ..AdvancedHMC - - import ..AdvancedHMC: ∂H∂r, neg_energy, AbstractKinetic - import Random: AbstractRNG - include("relativistic/hamiltonian.jl") - export RelativisticKinetic, DimensionwiseRelativisticKinetic - - using AdaptiveRejectionSampling: RejectionSampler, run_sampler! - import ..AdvancedHMC: _rand - include("relativistic/metric.jl") - - using ..AdvancedHMC: @unpack, TYPEDEF, TYPEDFIELDS, AbstractScalarOrVec, AbstractLeapfrog, step, step_size - import ..AdvancedHMC: ∂H∂θ, ∂H∂r, DualValue, PhasePoint, phasepoint, step - include("riemannian/integrator.jl") - export GeneralizedLeapfrog - - import AdvancedHMC: _rand - using AdvancedHMC: AbstractMetric, PhasePoint - using LinearAlgebra: eigen, cholesky, Symmetric, Diagonal - include("riemannian/metric.jl") - export DenseRiemannianMetric - - import AdvancedHMC: DualValue, phasepoint, neg_energy, ∂H∂θ, ∂H∂r - using LinearAlgebra: logabsdet, tr - include("riemannian/hamiltonian.jl") +using Random, Statistics, LinearAlgebra +using ..AdvancedHMC + +import ..AdvancedHMC: ∂H∂r, neg_energy, AbstractKinetic +import Random: AbstractRNG +include("relativistic/hamiltonian.jl") +export RelativisticKinetic, DimensionwiseRelativisticKinetic + +using AdaptiveRejectionSampling: RejectionSampler, run_sampler! +import ..AdvancedHMC: _rand +include("relativistic/metric.jl") + +using ..AdvancedHMC: + @unpack, TYPEDEF, TYPEDFIELDS, AbstractScalarOrVec, AbstractLeapfrog, step, step_size +import ..AdvancedHMC: ∂H∂θ, ∂H∂r, DualValue, PhasePoint, phasepoint, step +include("riemannian/integrator.jl") +export GeneralizedLeapfrog + +import AdvancedHMC: _rand +using AdvancedHMC: AbstractMetric, PhasePoint +using LinearAlgebra: eigen, cholesky, Symmetric, Diagonal +include("riemannian/metric.jl") +export DenseRiemannianMetric + +import AdvancedHMC: DualValue, phasepoint, neg_energy, ∂H∂θ, ∂H∂r +using LinearAlgebra: logabsdet, tr +include("riemannian/hamiltonian.jl") end include("abstractmcmc.jl")