Remove CoupledNODE from deps

Author: SCiarella

Project.toml

@@ -6,7 +6,6 @@ version = "0.1.0"
 [deps]
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
-CoupledNODE = "88291d29-22ea-41b1-bc0b-03785bffce48"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
 LuxCore = "bb33d45b-7691-41d6-9220-0943567d0623"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"

test/Project.toml

@@ -1,7 +1,6 @@
 [deps]
 AttentionLayer = "3ee63b08-73c5-50c8-acc9-f395aa68c39a"
 ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
-CoupledNODE = "88291d29-22ea-41b1-bc0b-03785bffce48"
 Lux = "b2108857-7c20-44ae-9111-449ecde12c47"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 TestItemRunner = "f8b46487-2199-4994-9208-9a1283c18c0a"

test/test-model.jl

@@ -2,7 +2,6 @@ using Test
 using Lux
 using AttentionLayer: attention
 using ComponentArrays: ComponentArray
-using CoupledNODE: cnn
 using Random
 using Zygote: Zygote
 
@@ -19,16 +18,22 @@ using Zygote: Zygote
     rng = Xoshiro(123)
 
     # Test CNN Layer Setup
-    CnnLayers, _, _ = cnn(;
-        T = T,
-        D = D,
-        data_ch = 2 * D,                     # Input channels for CNN after concatenation
-        radii = [3, 3],
-        channels = [2, 2],
-        activations = [tanh, identity],
-        use_bias = [false, false],
-        rng,
-    )
+    r =
+        [3, 3], c =
+            [2, 2], σ =
+                [tanh, identity], b =
+                    [false, false], CnnLayers = (
+                        (
+                            Lux.Conv(
+                                ntuple(α -> 2r[i] + 1, D),
+                                c[i] => c[i+1],
+                                σ[i];
+                                use_bias = b[i],
+                                init_weight = glorot_uniform_T,
+                                pad = (ntuple(α -> 2r[i] + 1, D) .- 1) .÷ 2,
+                            ) for i in eachindex(r)
+                        )...,
+                    )
 
     # Verify CNN layers' setup
     @test CnnLayers != nothing