Skip to content

Commit a9551de

Browse files
committed
Merge branch 'master' into development
2 parents 8053626 + 129d6c2 commit a9551de

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

PyNN/InputsTest.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,14 @@
5151
data = pop.get_data('v', gather=False)
5252
analogsignal = data.segments[0].analogsignals[0]
5353
name = analogsignal.name
54-
source_ids = analogsignal.annotations['source_ids']
54+
55+
def get_source_ids_as(analogsignal):
56+
if 'source_ids' in analogsignal.annotations:
57+
return analogsignal.annotations['source_ids']
58+
elif 'channel_ids' in analogsignal.annotations: # See https://github.com/NeuralEnsemble/PyNN/pull/762
59+
return analogsignal.annotations['channel_ids']
60+
61+
source_ids = get_source_ids_as(analogsignal)
5562
filename = "%s_%s.dat"%(pop.label,name)
5663
print('Saving data recorded for %s in pop %s, global ids: %s to %s'%(name, pop.label, source_ids, filename))
5764
times_vm_a = []
@@ -94,7 +101,7 @@ def get_source_id(spiketrain):
94101
sim.end()
95102

96103
if '-gui' in sys.argv:
97-
if simulator_name in ['neuron', 'nest', 'brian']:
104+
if simulator_name in ['neuron', 'nest', 'brian2']:
98105
import matplotlib.pyplot as plt
99106

100107
print("Plotting results of simulation in %s"%simulator_name)

0 commit comments

Comments
 (0)