Merge branch 'master' of https://github.com/ayushinav/NeuralPDE.jl

SciML · Mar 4, 2024 · 6189427 · 6189427
2 parents a867862 + e68f778
commit 6189427
Show file tree

Hide file tree

Showing 5 changed files with 239 additions and 239 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -25,6 +25,7 @@ jobs:
           - AdaptiveLoss
           - Logging
           - Forward
+          - NeuralAdapter
           - DGM
         version:
           - "1"
@@ -50,6 +51,8 @@ jobs:
       - uses: julia-actions/julia-processcoverage@v1
         with:
           directories: src,lib/NeuralPDELogging/src
-      - uses: codecov/codecov-action@v3
+      - uses: codecov/codecov-action@v4
         with:
           files: lcov.info
+          token: ${{ secrets.CODECOV_TOKEN }}
+          fail_ci_if_error: true
diff --git a/.github/workflows/Downstream.yml b/.github/workflows/Downstream.yml
@@ -48,6 +48,8 @@ jobs:
             exit(0)  # Exit immediately, as a success
           end
       - uses: julia-actions/julia-processcoverage@v1
-      - uses: codecov/codecov-action@v3
+      - uses: codecov/codecov-action@v4
         with:
           file: lcov.info
+          token: ${{ secrets.CODECOV_TOKEN }}
+          fail_ci_if_error: true
diff --git a/src/neural_adapter.jl b/src/neural_adapter.jl
@@ -28,11 +28,12 @@ function get_bounds_(domains, eqs, eltypeθ, dict_indvars, dict_depvars, strateg
                       for d in domains])
     args = get_argument(eqs, dict_indvars, dict_depvars)
 
-    bounds = map(args) do pd
+    bounds = first(map(args) do pd
         span = map(p -> get(dict_span, p, p), pd)
         map(s -> adapt(eltypeθ, s), span)
-    end
-    bounds
+    end)
+    bounds = [getindex.(bounds, 1), getindex.(bounds, 2)]
+    return bounds
 end
 
 function get_loss_function_(loss, init_params, pde_system, strategy::StochasticTraining)
@@ -46,8 +47,7 @@ function get_loss_function_(loss, init_params, pde_system, strategy::StochasticT
                                                             pde_system.depvars)
 
     eltypeθ = eltype(init_params)
-    bound = get_bounds_(domains, eqs, eltypeθ, dict_indvars, dict_depvars, strategy)[1]
-
+    bound = get_bounds_(domains, eqs, eltypeθ, dict_indvars, dict_depvars, strategy)
     get_loss_function(loss, bound, eltypeθ, strategy)
 end
 
@@ -62,8 +62,7 @@ function get_loss_function_(loss, init_params, pde_system, strategy::QuasiRandom
                                                             pde_system.depvars)
 
     eltypeθ = eltype(init_params)
-    bound = get_bounds_(domains, eqs, eltypeθ, dict_indvars, dict_depvars, strategy)[1]
-
+    bound = get_bounds_(domains, eqs, eltypeθ, dict_indvars, dict_depvars, strategy)
     get_loss_function(loss, bound, eltypeθ, strategy)
 end
 

diff --git a/src/training_strategies.jl b/src/training_strategies.jl
@@ -130,7 +130,6 @@ end
 function get_loss_function(loss_function, bound, eltypeθ, strategy::StochasticTraining;
                            τ = nothing)
     points = strategy.points
-
     loss = (θ) -> begin
         sets = generate_random_points(points, bound, eltypeθ)
         sets_ = adapt(parameterless_type(ComponentArrays.getdata(θ)), sets)