Merge pull request #30 from karandwivedi42/patch-1HEADmaster

Update Edge.lua

1 files changed, 2 insertions, 1 deletions

diff --git a/Edge.lua b/Edge.lua
index 1ffd670..bd99af2 100644
--- a/Edge.lua
+++ b/Edge.lua
@@ -4,7 +4,8 @@ No methods, just two fields, from and to.
 ]]--
 local Edge = torch.class('graph.Edge')
 
-function Edge:__init(from,to)
+function Edge:__init(from,to,weight)
    self.from = from
    self.to = to
+   self.weight = weight
 end