Skip to content

Commit 12d09eb

Browse files
authored
Flux no longer exports params
1 parent 9b16e02 commit 12d09eb

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

test/fourierlayer.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,13 @@ using Test, Random, Flux
1313
# We only use a subset of the weight tensors for training
1414
@testset "parameters" begin
1515
# Wf
16-
@test size(params(FourierLayer(128, 64, 100, 20))[1]) == (128, 64, 20)
16+
@test size(Flux.params(FourierLayer(128, 64, 100, 20))[1]) == (128, 64, 20)
1717
# Wl
18-
@test size(params(FourierLayer(128, 64, 100, 20))[2]) == (64, 128)
18+
@test size(Flux.params(FourierLayer(128, 64, 100, 20))[2]) == (64, 128)
1919
# bf
20-
@test size(params(FourierLayer(128, 64, 100, 20))[3]) == (1, 64, 20)
20+
@test size(Flux.params(FourierLayer(128, 64, 100, 20))[3]) == (1, 64, 20)
2121
# bl
22-
@test size(params(FourierLayer(128, 64, 100, 20))[4]) == (1, 64, 100)
22+
@test size(Flux.params(FourierLayer(128, 64, 100, 20))[4]) == (1, 64, 100)
2323
end
2424

2525
# Accept only Int as architecture parameters
@@ -28,4 +28,4 @@ using Test, Random, Flux
2828
# Test max amount of modes
2929
@test_throws AssertionError FourierLayer(100, 100, 100, 60, σ)
3030
@test_throws AssertionError FourierLayer(100, 100, 100, 60)
31-
end
31+
end

0 commit comments

Comments
 (0)