Sequential:insert is more general

1 files changed, 7 insertions, 6 deletions

diff --git a/Sequential.lua b/Sequential.lua
index bdd3ac4..ec3247b 100644
--- a/Sequential.lua
+++ b/Sequential.lua
@@ -13,13 +13,14 @@ function Sequential:add(module)
    return self
 end
 
-function Sequential:push(module)
-   local modules = self.modules
-   self.modules = {}
-   self:add(module)
-   for i,module in ipairs(modules) do
-      self:add(module)
+function Sequential:insert(module, index)
+   index = index or (#self.modules + 1)
+   if index > (#self.modules + 1) then
+      error"index should be contiguous to existing modules"
    end
+   table.insert(self.modules, index, module)
+   self.output = self.modules[#self.modules].output
+   self.gradInput = self.modules[1].gradInput
 end
 
 function Sequential:size()