Type issue in Julia v1.9

[jiyess]

I recently began working through the examples provided in the tutorial in Julia v1.9. I noticed that the examples for Ordinary Differential Equations (ODEs) run smoothly without any issues. However, when I attempted to replicate the examples for Partial Differential Equations (PDEs), I encountered some type errors. I've included the relevant error messages below for your reference. Could you possibly provide some guidance on resolving these issues?

ERROR: TypeError: in NonAdaptiveLoss, in T, expected T<:Real, got Type{Any}
Stacktrace:
 [1] symbolic_discretize(pde_system::PDESystem, discretization::PhysicsInformedNN{GridTraining{Float64}, Nothing, NeuralPDE.Phi{Lux.Chain{NamedTuple{(:layer_1, :layer_2, :layer_3), Tuple{Dense{typeof(σ), Matrix{Float32}, Vector{Float32}}, Dense{typeof(σ), Matrix{Float32}, Vector{Float32}}, Dense{typeof(identity), Matrix{Float32}, Vector{Float32}}}}, Nothing}, NamedTuple{(:layer_1, :layer_2, :layer_3), Tuple{NamedTuple{(), Tuple{}}, NamedTuple{(), Tuple{}}, NamedTuple{(), Tuple{}}}}}, typeof(NeuralPDE.numeric_derivative), Bool, Nothing, Nothing, Nothing, Base.Pairs{Symbol, Union{}, Tuple{}, NamedTuple{(), Tuple{}}}})
   @ NeuralPDE ~/.julia/packages/NeuralPDE/gdbjo/src/discretize.jl:490
 [2] discretize(pde_system::PDESystem, discretization::PhysicsInformedNN{GridTraining{Float64}, Nothing, NeuralPDE.Phi{Lux.Chain{NamedTuple{(:layer_1, :layer_2, :layer_3), Tuple{Dense{typeof(σ), Matrix{Float32}, Vector{Float32}}, Dense{typeof(σ), Matrix{Float32}, Vector{Float32}}, Dense{typeof(identity), Matrix{Float32}, Vector{Float32}}}}, Nothing}, NamedTuple{(:layer_1, :layer_2, :layer_3), Tuple{NamedTuple{(), Tuple{}}, NamedTuple{(), Tuple{}}, NamedTuple{(), Tuple{}}}}}, typeof(NeuralPDE.numeric_derivative), Bool, Nothing, Nothing, Nothing, Base.Pairs{Symbol, Union{}, Tuple{}, NamedTuple{(), Tuple{}}}})
   @ NeuralPDE ~/.julia/packages/NeuralPDE/gdbjo/src/discretize.jl:683
 [3] top-level scope
   @ ~/Documents/research/research/PINN_sol/PDE_test1:28

[ChrisRackauckas]

This is indicative of running an old version. Please make sure you're using NeuralPDE v5.8.