fix: BPINN ODE testing

SciML · Oct 13, 2024 · 648b251 · 648b251
1 parent 9bd5e28
commit 648b251
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/advancedHMC_MCMC.jl b/src/advancedHMC_MCMC.jl
@@ -353,8 +353,8 @@ NN OUTPUT AT t,θ ~ phi(t,θ).
 """
 function (f::LogTargetDensity{C, S})(t::AbstractVector,
  θ) where {C <: Lux.AbstractLuxLayer, S}
- θ = vector_to_parameters(θ, f.init_params)
  eltypeθ, typeθ = eltype(θ), parameterless_type(ComponentArrays.getdata(θ))
+ θ = vector_to_parameters(θ, f.init_params)
  t_ = convert.(eltypeθ, adapt(typeθ, t'))
  y, st = f.chain(t_, θ, f.st)
  ChainRulesCore.@ignore_derivatives f.st = st
@@ -363,8 +363,8 @@ end
 
 function (f::LogTargetDensity{C, S})(t::Number,
  θ) where {C <: Lux.AbstractLuxLayer, S}
- θ = vector_to_parameters(θ, f.init_params)
  eltypeθ, typeθ = eltype(θ), parameterless_type(ComponentArrays.getdata(θ))
+ θ = vector_to_parameters(θ, f.init_params)
  t_ = convert.(eltypeθ, adapt(typeθ, [t]))
  y, st = f.chain(t_, θ, f.st)
  ChainRulesCore.@ignore_derivatives f.st = st

diff --git a/test/BPINN_Tests.jl b/test/BPINN_Tests.jl
@@ -1,4 +1,4 @@
-# # Testing Code
+# Testing Code
 using Test, MCMCChains
 using ForwardDiff, Distributions, OrdinaryDiffEq
 using OptimizationOptimisers, AdvancedHMC, Lux