update NeuralPDE.jl, advancedHMC_MCMC.jl

SciML · Mar 26, 2024 · 9f694f8 · 9f694f8
1 parent 16e1b56
commit 9f694f8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/src/NeuralPDE.jl b/src/NeuralPDE.jl
@@ -53,7 +53,6 @@ include("advancedHMC_MCMC.jl")
 include("BPINN_ode.jl")
 include("PDE_BPINN.jl")
 include("dgm.jl")
-include("collocated_estim.jl")
 
 export NNODE, NNDAE,
     PhysicsInformedNN, discretize,

diff --git a/src/advancedHMC_MCMC.jl b/src/advancedHMC_MCMC.jl
@@ -448,8 +448,7 @@ function ahmc_bayesian_pinn_ode(prob::DiffEqBase.ODEProblem, chain;
     progress = false, verbose = false,
     estim_collocate = false)
 
-    !(chain isa Lux.AbstractExplicitLayer) &&
-        (chain = adapt(FromFluxAdaptor(false, false), chain))
+    !(chain isa Lux.AbstractExplicitLayer) && (chain = adapt(FromFluxAdaptor(false, false), chain))
     # NN parameter prior mean and variance(PriorsNN must be a tuple)
     if isinplace(prob)
         throw(error("The BPINN ODE solver only supports out-of-place ODE definitions, i.e. du=f(u,p,t)."))