Skip to content

Commit ef98a97

Browse files
author
Nicholas Leonard
committed
fix MaskZero unit test name
1 parent f911b83 commit ef98a97

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test.lua

+2-2
Original file line numberDiff line numberDiff line change
@@ -4188,7 +4188,7 @@ local function firstElement(a)
41884188
return torch.type(a) == 'table' and a[1] or a
41894189
end
41904190

4191-
function rnntest.MaskZero()
4191+
function rnntest.MaskZero_main()
41924192
local recurrents = {['recurrent'] = recurrentModule(), ['lstm'] = lstmModule()}
41934193
-- Note we use lstmModule input signature and firstElement to prevent duplicate code
41944194
for name, recurrent in pairs(recurrents) do
@@ -4290,7 +4290,7 @@ function rnntest.MaskZero()
42904290
end
42914291
end
42924292

4293-
function rnntest.TrimZero()
4293+
function rnntest.TrimZero_main()
42944294
local recurrents = {['recurrent'] = recurrentModule(), ['lstm'] = lstmModule()}
42954295
-- Note we use lstmModule input signature and firstElement to prevent duplicate code
42964296
for name, recurrent in pairs(recurrents) do

0 commit comments

Comments
 (0)