diff --git a/hnn_core/hnn_io.py b/hnn_core/hnn_io.py index 5344115c4..fe9625a29 100644 --- a/hnn_core/hnn_io.py +++ b/hnn_core/hnn_io.py @@ -177,6 +177,7 @@ def _read_external_drive(net, drive_data, read_output): sigma=drive_data['dynamics']['sigma'], numspikes=drive_data['dynamics']['numspikes'], location=drive_data['location'], + n_drive_cells=_set_from_cell_specific(drive_data), cell_specific=drive_data['cell_specific'], weights_ampa=drive_data['weights_ampa'], weights_nmda=drive_data['weights_nmda'],