Skip to content

Commit aeb421b

Browse files
fix test
1 parent f288f3f commit aeb421b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/utils.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -469,10 +469,10 @@ end
469469
@test modules[5] === m2
470470
@test modules[6] === m3
471471

472-
mod_par = Flux.modules(Parallel(Flux.Bilinear(2,2,2,cbrt), Dense(2,2,abs), Dense(2,2,abs2)))
472+
mod_par = Flux.modules(Parallel(Flux.Bilinear(2,2,2,cbrt), Dense(2=>2,abs), Dense(2=>2,abs2)))
473473
@test length(mod_par) == 5
474474

475-
mod_rnn = Flux.modules(Chain(Dense(2,3), BatchNorm(3), LSTM(3,4)))
475+
mod_rnn = Flux.modules(Chain(Dense(2=>3), BatchNorm(3), LSTM(3=>4)))
476476
@test length(mod_rnn) == 6
477477
@test mod_rnn[end] isa Flux.LSTMCell
478478

0 commit comments

Comments
 (0)