Updated for latest _get_spiketimes (#598)

NeuralEnsemble · Oct 4, 2018 · bf87b98 · bf87b98
1 parent 6e7cdf8
commit bf87b98
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pyNN/neuroml/recording.py b/pyNN/neuroml/recording.py
@@ -67,7 +67,14 @@ def _record(self, variable, new_ids, sampling_interval=None):
 
 
     def _get_spiketimes(self, id):
-        return numpy.array([id, id+5], dtype=float) % self._simulator.state.t
+
+        if hasattr(id, "__len__"):
+            spks = {}
+            for i in id:
+                spks[i] = numpy.array([i, i + 5], dtype=float) % self._simulator.state.t
+            return spks
+        else:
+            return numpy.array([id, id + 5], dtype=float) % self._simulator.state.t
 
     def _get_all_signals(self, variable, ids, clear=False):
         # assuming not using cvode, otherwise need to get times as well and use IrregularlySampledAnalogSignal