fix

JuliaML · Feb 4, 2025 · 146f600 · 146f600
1 parent 7a6bcbb
commit 146f600
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/test/batch.jl b/test/batch.jl
@@ -78,13 +78,13 @@ end
     @test bs[4] == [1.0 0.0 0.0; 1.0 0.0 0.0]
 end
 
-@test "batch_sequences" begin
-    y = batchseqs([[1, 2, 3], [10, 20]])
+@testset "batch_sequences" begin
+    y = batch_sequence([[1, 2, 3], [10, 20]])
     @test y isa Matrix{Int}
     @test y == [1 10; 2 20; 3 0]
 
     data = (ones(2, 1), fill(2.0, (2, 3)))
-    y = batchseqs(data, pad=-1)
+    y = batch_sequence(data, pad=-1)
     @test y[:,:,1] == [1 -1 -1; 1 -1 -1]
     @test y[:,:,2] == [2 2 2; 2 2 2]
 end
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -89,6 +89,8 @@ include("test_utils.jl")
 
 # @testset "MLUtils.jl" begin
 
+
+@testset "batch" begin; include("batch.jl"); end
 @testset "batchview" begin; include("batchview.jl"); end
 @testset "dataloader" begin; include("dataloader.jl"); end
 @testset "folds" begin; include("folds.jl"); end