Update src/AdvancedHMC.jl

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
TuringLang · Jul 26, 2024 · 8805885 · 8805885
1 parent 3813ba0
commit 8805885
Showing 1 changed file with 27 additions and 26 deletions.
diff --git 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")