fix MaskZero unit test name

torch · Feb 20, 2017 · ef98a97 · ef98a97
1 parent f911b83
commit ef98a97
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test.lua b/test/test.lua
@@ -4188,7 +4188,7 @@ local function firstElement(a)
  return torch.type(a) == 'table' and a[1] or a
 end
 
-function rnntest.MaskZero()
+function rnntest.MaskZero_main()
  local recurrents = {['recurrent'] = recurrentModule(), ['lstm'] = lstmModule()}
  -- Note we use lstmModule input signature and firstElement to prevent duplicate code
  for name, recurrent in pairs(recurrents) do
@@ -4290,7 +4290,7 @@ function rnntest.MaskZero()
  end
 end
 
-function rnntest.TrimZero()
+function rnntest.TrimZero_main()
  local recurrents = {['recurrent'] = recurrentModule(), ['lstm'] = lstmModule()}
  -- Note we use lstmModule input signature and firstElement to prevent duplicate code
  for name, recurrent in pairs(recurrents) do