|
1 | 1 | from neuromllite import Network, Cell, InputSource, Population, Synapse |
2 | | -from neuromllite import Projection, RandomConnectivity, Input, Simulation, RectangularRegion, RandomLayout |
| 2 | +from neuromllite import ( |
| 3 | + Projection, |
| 4 | + RandomConnectivity, |
| 5 | + Input, |
| 6 | + Simulation, |
| 7 | + RectangularRegion, |
| 8 | + RandomLayout, |
| 9 | +) |
3 | 10 | import sys |
4 | 11 |
|
5 | 12 | ################################################################################ |
|
11 | 18 | net.seed = 1234 |
12 | 19 | net.temperature = 32.0 |
13 | 20 |
|
14 | | -net.parameters = {"N": 20, |
15 | | - "fractionE": 0.7, |
16 | | - "weightInput": 0.7, |
17 | | - "prob_e_e": 0.1, |
18 | | - "prob_e_i": 0.9, |
19 | | - "prob_i_e": 0.8, |
20 | | - "prob_i_i": 0.3, |
21 | | - "global_delay": 2} |
22 | | - |
23 | | -r1 = RectangularRegion( |
24 | | - id="region1", x=0, y=0, z=0, width=1000, height=100, depth=1000 |
25 | | -) |
| 21 | +net.parameters = { |
| 22 | + "N": 20, |
| 23 | + "fractionE": 0.7, |
| 24 | + "weightInput": 0.7, |
| 25 | + "prob_e_e": 0.1, |
| 26 | + "prob_e_i": 0.9, |
| 27 | + "prob_i_e": 0.8, |
| 28 | + "prob_i_i": 0.3, |
| 29 | + "global_delay": 2, |
| 30 | +} |
| 31 | + |
| 32 | +r1 = RectangularRegion(id="region1", x=0, y=0, z=0, width=1000, height=100, depth=1000) |
26 | 33 | net.regions.append(r1) |
27 | 34 |
|
28 | | -pyr_cell = Cell(id="pyr_4_sym", neuroml2_source_file="test_files/acnet2/pyr_4_sym.cell.nml") |
| 35 | +pyr_cell = Cell( |
| 36 | + id="pyr_4_sym", neuroml2_source_file="test_files/acnet2/pyr_4_sym.cell.nml" |
| 37 | +) |
29 | 38 | net.cells.append(pyr_cell) |
30 | 39 | bask_cell = Cell(id="bask", neuroml2_source_file="test_files/acnet2/bask.cell.nml") |
31 | 40 | net.cells.append(bask_cell) |
|
61 | 70 | net.populations.append(pE) |
62 | 71 | net.populations.append(pI) |
63 | 72 |
|
64 | | -syn_e_e = Synapse(id="AMPA_syn", neuroml2_source_file="test_files/acnet2/AMPA_syn.synapse.nml") |
| 73 | +syn_e_e = Synapse( |
| 74 | + id="AMPA_syn", neuroml2_source_file="test_files/acnet2/AMPA_syn.synapse.nml" |
| 75 | +) |
65 | 76 | net.synapses.append(syn_e_e) |
66 | | -syn_e_i = Synapse(id="AMPA_syn_inh", neuroml2_source_file="test_files/acnet2/AMPA_syn_inh.synapse.nml") |
| 77 | +syn_e_i = Synapse( |
| 78 | + id="AMPA_syn_inh", neuroml2_source_file="test_files/acnet2/AMPA_syn_inh.synapse.nml" |
| 79 | +) |
67 | 80 | net.synapses.append(syn_e_i) |
68 | | -syn_i_e = Synapse(id="GABA_syn", neuroml2_source_file="test_files/acnet2/GABA_syn.synapse.nml") |
| 81 | +syn_i_e = Synapse( |
| 82 | + id="GABA_syn", neuroml2_source_file="test_files/acnet2/GABA_syn.synapse.nml" |
| 83 | +) |
69 | 84 | net.synapses.append(syn_i_e) |
70 | | -syn_i_i = Synapse(id="GABA_syn_inh", neuroml2_source_file="test_files/acnet2/GABA_syn_inh.synapse.nml") |
| 85 | +syn_i_i = Synapse( |
| 86 | + id="GABA_syn_inh", neuroml2_source_file="test_files/acnet2/GABA_syn_inh.synapse.nml" |
| 87 | +) |
71 | 88 | net.synapses.append(syn_i_i) |
72 | 89 |
|
73 | 90 |
|
74 | | - |
75 | 91 | net.projections.append( |
76 | 92 | Projection( |
77 | 93 | id="projEE", |
|
94 | 110 | ) |
95 | 111 |
|
96 | 112 |
|
97 | | - |
98 | 113 | net.projections.append( |
99 | 114 | Projection( |
100 | 115 | id="projIE", |
|
136 | 151 | ### Build Simulation object & save as JSON |
137 | 152 |
|
138 | 153 | sim = Simulation( |
139 | | - id="Sim%s"%net.id, |
| 154 | + id="Sim%s" % net.id, |
140 | 155 | network=new_file, |
141 | 156 | duration="1000", |
142 | 157 | seed="1111", |
|
145 | 160 | record_spikes={ |
146 | 161 | pE.id: "*", |
147 | 162 | pI.id: "*", |
148 | | - } |
| 163 | + }, |
149 | 164 | ) |
150 | 165 |
|
151 | 166 | sim.to_json_file() |
|
0 commit comments