From 5232a3300c80a35ad1839fc6f163cf97459bbec8 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Fri, 2 Jun 2017 16:22:55 +0100
Subject: [PATCH] don't return hidden states

---
 src/compiler/loops.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/compiler/loops.jl b/src/compiler/loops.jl
index 4a34c3c7..576aeb4f 100644
--- a/src/compiler/loops.jl
+++ b/src/compiler/loops.jl
@@ -140,7 +140,7 @@ function stateless(s::Stateful)
   v = graph(s.model)
   v = spliceinputs(v, group(constant.(s.states)...),
                    [inputnode(i) for i = 1:graphinputs(v)-1]...)
-  Capacitor(v)
+  Capacitor(v[2])
 end
 
 stateless(s::SeqModel) = SeqModel(stateless(s.model), s.steps)