Merge pull request #50 from fidlej/topic_argument_check

Checked that gModule inputs and outputs are Nodes.

1 files changed, 10 insertions, 0 deletions

diff --git a/gmodule.lua b/gmodule.lua
index 8dc100d..e6e5fab 100644
--- a/gmodule.lua
+++ b/gmodule.lua
@@ -50,8 +50,18 @@ function gModule:__init(inputs,outputs)
 	-- input point for the backward graph
 	local outnode = nngraph.Node({input={}})
 	for i,n in ipairs(outputs) do
+		if torch.typename(n) ~= 'nngraph.Node' then
+			error(string.format('what is this in the outputs[%s]? %s',
+				i, tostring(n)))
+		end
 		outnode:add(n,true)
 	end
+	for i,n in ipairs(inputs) do
+		if torch.typename(n) ~= 'nngraph.Node' then
+			error(string.format('what is this in the inputs[%s]? %s',
+				i, tostring(n)))
+		end
+	end
 	-- We add also a dummy input node.
 	-- The input node will be split to feed the passed input nodes.
 	local innode = nngraph.Node({input={}})