Fix for training on latest version

.github/workflows/CompatHelper.yml

@@ -0,0 +1,16 @@
+name: CompatHelper
+on:
+  schedule:
+    - cron: 0 0 * * *
+  workflow_dispatch:
+jobs:
+  CompatHelper:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Pkg.add("CompatHelper")
+        run: julia -e 'using Pkg; Pkg.add("CompatHelper")'
+      - name: CompatHelper.main()
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          COMPATHELPER_PRIV: ${{ secrets.DOCUMENTER_KEY }}
+        run: julia -e 'using CompatHelper; CompatHelper.main()'

src/experiments/models_gnn.jl

@@ -51,7 +51,7 @@ struct MPSolver{E, P, D, T<: AbstractFloat} <:
     encoder::E
     processor::P
     decoder::D
-    Δt::Vector{T}
+    Δt::AbstractArray{T}
 end
 
 """

src/experiments/utilis.jl

@@ -2,7 +2,7 @@ function get_dataloader(args)
     e = args.experiment
     !isfile("datasets/$e.jld2") && generate_save_data(e) #TODO: add wave equation in the future
 
-    (;domain, u, dx, dt, θ) = load("datasets/$e.jld2")
+    (domain, u, dx, dt, θ) = values(load("datasets/$e.jld2"))
 
     u = u[:,1:2:end,:] # downsample
     dx = dx * 2

src/generate/generate_data_CE.jl

@@ -96,7 +96,7 @@ function generate_data(
                 rand(Uniform(0, 2π), 5),
             ),
         )
-        u[:, :, i] .= Array{T}(solve(newprob, Tsit5(), saveat = dt))'
+        u[:, :, i] .= Array{T}(solve(newprob, Tsit5(), saveat = dt).original_sol)'
     end
     return u, dx, dt, θ, domain
 end