Skip to content

Commit 52d920e

Browse files
authored
Merge branch 'master' into clone_connectors
2 parents 29ab6b1 + d315772 commit 52d920e

4 files changed

Lines changed: 9 additions & 6 deletions

File tree

spynnaker/pyNN/models/neuron/synapse_io.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -385,7 +385,8 @@ def _get_row_data(
385385
def convert_to_connections(
386386
synapse_info: SynapseInformation, post_vertex_slice: Slice,
387387
n_pre_atoms: int, max_row_length: int, n_synapse_types: int,
388-
weight_scales: WeightScales, data: Union[bytes, NDArray, None],
388+
weight_scales: WeightScales,
389+
data: bytearray | bytes | memoryview | NDArray | None,
389390
delayed: bool, post_vertex_max_delay_ticks: int,
390391
max_atoms_per_core: int) -> ConnectionsArray:
391392
"""

spynnaker/pyNN/models/neuron/synaptic_matrix_app.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -405,6 +405,7 @@ def get_connections(self, placement: Placement) -> List[NDArray]:
405405
splitter = self.__app_edge.post_vertex.splitter
406406
vertex_slice = placement.vertex.vertex_slice
407407
if self.__syn_mat_offset is not None:
408+
block: bytearray | bytes | memoryview
408409
if self.__download_index is not None:
409410
assert buffers is not None
410411
block, _ = buffers.get_download(
@@ -436,7 +437,7 @@ def get_connections(self, placement: Placement) -> List[NDArray]:
436437
return connections
437438

438439
def __get_block(
439-
self, placement: Placement, synapses_address: int) -> bytes:
440+
self, placement: Placement, synapses_address: int) -> bytearray:
440441
"""
441442
Get a block of data for undelayed synapses.
442443
@@ -451,7 +452,7 @@ def __get_block(
451452
placement.x, placement.y, address, self.__matrix_size)
452453

453454
def __get_delayed_block(
454-
self, placement: Placement, synapses_address: int) -> bytes:
455+
self, placement: Placement, synapses_address: int) -> bytearray:
455456
"""
456457
Get a block of data for delayed synapses.
457458

spynnaker/pyNN/utilities/struct.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -332,7 +332,8 @@ def is_generatable(self) -> bool:
332332
for data_type, _name in self.__fields)
333333

334334
def read_data(
335-
self, data: bytes, values: RangeDictionary, data_offset: int = 0,
335+
self, data: bytearray | bytes, values: RangeDictionary,
336+
data_offset: int = 0,
336337
vertex_slice: Optional[Slice] = None) -> None:
337338
"""
338339
Read a byte string of data and write to values.

unittests/model_tests/neuron/test_synaptic_manager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
# limitations under the License.
1414
import shutil
1515
import struct
16-
from typing import Any, BinaryIO, List, Optional, Sequence, Tuple, Union
16+
from typing import Any, BinaryIO, List, Optional, Sequence, Tuple
1717
import unittest
1818
from tempfile import mkdtemp
1919
import numpy
@@ -71,7 +71,7 @@ def malloc_sdram(
7171
@overrides(MockableTransceiver.write_memory)
7272
def write_memory(
7373
self, x: int, y: int, base_address: int,
74-
data: Union[BinaryIO, bytes, int, str], *,
74+
data: BinaryIO | bytearray | bytes | int | str, *,
7575
n_bytes: Optional[int] = None, offset: int = 0, cpu: int = 0,
7676
get_sum: bool = False) -> Tuple[int, int]:
7777
if data is None:

0 commit comments

Comments
 (0)