fix test

FluxML · Oct 17, 2024 · aeb421b · aeb421b
1 parent f288f3f
commit aeb421b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/utils.jl b/test/utils.jl
@@ -469,10 +469,10 @@ end
   @test modules[5] === m2
   @test modules[6] === m3
 
-  mod_par = Flux.modules(Parallel(Flux.Bilinear(2,2,2,cbrt), Dense(2,2,abs), Dense(2,2,abs2)))
+  mod_par = Flux.modules(Parallel(Flux.Bilinear(2,2,2,cbrt), Dense(2=>2,abs), Dense(2=>2,abs2)))
   @test length(mod_par) == 5
 
-  mod_rnn = Flux.modules(Chain(Dense(2,3), BatchNorm(3), LSTM(3,4)))
+  mod_rnn = Flux.modules(Chain(Dense(2=>3), BatchNorm(3), LSTM(3=>4)))
   @test length(mod_rnn) == 6
   @test mod_rnn[end] isa Flux.LSTMCell