Skip to content

Commit

Permalink
Fix changes for better pickling of Synapses
Browse files Browse the repository at this point in the history
  • Loading branch information
mstimberg committed Jul 25, 2023
1 parent 01b6c63 commit d3d9e16
Showing 1 changed file with 9 additions and 5 deletions.
14 changes: 9 additions & 5 deletions brian2/synapses/synapses.py
Original file line number Diff line number Diff line change
Expand Up @@ -980,10 +980,11 @@ def __init__(
self._pathways = []

if isinstance(on_event, str):
events_dict = {"pre": on_event, "post": on_event}
self.default_event = on_event
events_dict = {}
else:
events_dict = {"pre": "spike", "post": "spike"}
events_dict.update(on_event)
self.default_event = "spike"
events_dict = dict(on_event)

#: "Events" for all the pathways
self.events = events_dict
Expand All @@ -993,7 +994,10 @@ def __init__(
if isinstance(argument, str):
pathway_delay = delay.get(prepost, None)
self._add_updater(
argument, prepost, delay=pathway_delay, event=self.events[prepost]
argument,
prepost,
delay=pathway_delay,
event=self.events.get(prepost, self.default_event),
)
elif isinstance(argument, Mapping):
for key, value in argument.items():
Expand All @@ -1010,7 +1014,7 @@ def __init__(
prepost,
objname=key,
delay=pathway_delay,
event=self.events[key],
event=self.events.get(prepost, self.default_event),
)

# Check whether any delays were specified for pathways that don't exist
Expand Down

0 comments on commit d3d9e16

Please sign in to comment.