Skip to content

Commit 05fcb99

Browse files
committed
Fix the failing CI build
1 parent f3160be commit 05fcb99

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/utils.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
using Flux
22
using Flux: throttle, nfan, glorot_uniform, glorot_normal,
33
kaiming_normal, kaiming_uniform, orthogonal, truncated_normal,
4-
sparse_init, identity_init, stack, unstack, batch, unbatch,
4+
sparse_init, identity_init, unstack, batch, unbatch,
55
unsqueeze, params, loadparams!, loadmodel!
66
using StatsBase: var, std
77
using Statistics, LinearAlgebra
@@ -326,14 +326,14 @@ end
326326

327327
@testset "Stacking" begin
328328
x = randn(3,3)
329-
stacked = stack([x, x], dims=2)
329+
stacked = Flux.MLUtils.stack([x, x], dims=2)
330330
@test size(stacked) == (3,2,3)
331331

332332
stacked_array=[ 8 9 3 5; 9 6 6 9; 9 1 7 2; 7 4 10 6 ]
333333
unstacked_array=[[8, 9, 9, 7], [9, 6, 1, 4], [3, 6, 7, 10], [5, 9, 2, 6]]
334334
@test unstack(stacked_array, dims=2) == unstacked_array
335-
@test stack(unstacked_array, dims=2) == stacked_array
336-
@test stack(unstack(stacked_array, dims=1), dims=1) == stacked_array
335+
@test Flux.MLUtils.stack(unstacked_array, dims=2) == stacked_array
336+
@test Flux.MLUtils.stack(unstack(stacked_array, dims=1), dims=1) == stacked_array
337337
end
338338

339339
@testset "Batching" begin

0 commit comments

Comments
 (0)