From acad60447031c77ac2b771e3594835be00ca3d0d Mon Sep 17 00:00:00 2001
From: paramthakkar123 <paramthakkar864@gmail.com>
Date: Sat, 1 Feb 2025 20:37:46 +0530
Subject: [PATCH] Fixes

---
 Project.toml     | 1 +
 src/NeuralPDE.jl | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/Project.toml b/Project.toml
index b1e5bd7df..98c98c251 100644
--- 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
index 5810759d1..ba947fa7e 100644
--- 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