Debugging ActionSelection.

juancarlosfarah · May 18, 2015 · 810bf04 · 810bf04
1 parent 8d795c5
commit 810bf04
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/training/training/training.py b/src/training/training/training.py
@@ -127,12 +127,13 @@ def run(self):
         sim.run()
 
         # Adding ActionSelection
-        
+
         pattern_input = []
         for neuron in sim.neurons:
             pattern_input.append(neuron.spike_times)
 
-        a_s = ActionSelection(pattern_input=input, pattern_duration=len(input),
+        a_s = ActionSelection(pattern_input=input,
+                              pattern_duration=self.ani.total_frames,
                               animation=self.ani)
         a_s.run(show_plots=False)