From efbccda4297d4c89d1515ea943aee4df0b665bb7 Mon Sep 17 00:00:00 2001
From: Astitva Aggarwal <astitvaaggarwalimportant@gmail.com>
Date: Fri, 29 Mar 2024 23:47:36 +0530
Subject: [PATCH] update training_strategies.jl

---
 src/training_strategies.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/training_strategies.jl b/src/training_strategies.jl
index e33f490fe7..d128d7bb5f 100644
--- a/src/training_strategies.jl
+++ b/src/training_strategies.jl
@@ -48,7 +48,7 @@ end
 
 # include dataset points in pde_residual loglikelihood (BayesianPINN)
 function merge_strategy_with_loglikelihood_function(pinnrep::PINNRepresentation,
-        strategy::GridTraining,
+        strategy,
         datafree_pde_loss_function,
         datafree_bc_loss_function; train_sets_pde = nothing, train_sets_bc = nothing)
     @unpack domains, eqs, bcs, dict_indvars, dict_depvars, flat_init_params = pinnrep