update training_strategies.jl

SciML · Mar 29, 2024 · efbccda · efbccda
1 parent 3e96e3d
commit efbccda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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