Merge pull request #112 from rohanpadhye/module-type

Return type of gModule when no arguments are passed to method: type().

1 files changed, 5 insertions, 0 deletions

diff --git a/gmodule.lua b/gmodule.lua
index 2a367d4..e5bd4d4 100644
--- a/gmodule.lua
+++ b/gmodule.lua
@@ -220,6 +220,10 @@ local function recursiveType(param, type_str)
 end
 
 function gModule:type(type, tensorCache)
+   if not type then
+      return self._type
+   end
+
    tensorCache = tensorCache or {}
 
    local function applyTypeToTable(table)
@@ -244,6 +248,7 @@ function gModule:type(type, tensorCache)
       end
    end
 
+   self._type = type
    return self
 end