X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=dagnn.git;a=blobdiff_plain;f=test-dagnn.lua;h=262ea6fe3111830ab1f8270118b608725e124881;hp=a0a81ab0e146988b3bc7859d4a90366b03e57b1f;hb=452781856eafd237579e5c90b6e345354df91b42;hpb=be03a73e411d18082a2dd99bff5df45c085017ca

diff --git a/test-dagnn.lua b/test-dagnn.lua
index a0a81ab..262ea6f 100755
--- a/test-dagnn.lua
+++ b/test-dagnn.lua
@@ -21,10 +21,11 @@ f = nn.Linear(3, 2)
                      \---> f ---
 ]]--
 
-g = DAG:new()
+g = nn.DAG:new()
 
 g:setInput(a)
 g:setOutput({ e, f })
+
 g:addEdge(c, e)
 g:addEdge(a, b)
 g:addEdge(d, e)
@@ -32,9 +33,7 @@ g:addEdge(b, c)
 g:addEdge(b, d)
 g:addEdge(d, f)
 
-g:order()
-
-g:print(graph)
+g:print()
 
 input = torch.Tensor(3, 10):uniform()