Fixes

SciML · Feb 1, 2025 · acad604 · acad604
1 parent cc96f24
commit acad604
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -22,6 +22,7 @@ IntervalSets = "8197267c-284f-5f27-9208-e0e47529a953"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 LogDensityProblems = "6fdf6af0-433a-55f7-b3ed-c6c6e0b8df7c"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
+OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
 LuxCore = "bb33d45b-7691-41d6-9220-0943567d0623"
 MCMCChains = "c7f686f2-ff18-58e9-bc7b-31028e88f75d"
 MLDataDevices = "7e8f7934-dd98-4c1a-8fe8-92b47a384d40"

diff --git a/src/NeuralPDE.jl b/src/NeuralPDE.jl
@@ -77,6 +77,7 @@ include("pinn_types.jl")
 include("symbolic_utilities.jl")
 include("training_strategies.jl")
 include("adaptive_losses.jl")
+include("../lib/BayesianNeuralPDE/src/BayesianNeuralPDE.jl")
 
 include("ode_solve.jl")
 include("dae_solve.jl")
@@ -94,7 +95,7 @@ export NNODE, NNDAE
 export PhysicsInformedNN, discretize
 export DeepGalerkin
 
-using BayesianNeuralPDE
+using .BayesianNeuralPDE
 export ahmc_bayesian_pinn_ode, ahmc_bayesian_pinn_pde,
        BNNODE, BPINNsolution, BayesianPINN