Get 2d size correct

SpiNNakerManchester · Oct 12, 2024 · 151820f · 151820f
1 parent 6daa42a
commit 151820f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spynnaker/pyNN/models/neural_projections/connectors/kernel_connector.py b/spynnaker/pyNN/models/neural_projections/connectors/kernel_connector.py
@@ -523,9 +523,9 @@ def gen_connector_params(
     def gen_connector_params_size_in_bytes(self) -> int:
         size = N_KERNEL_PARAMS * BYTES_PER_WORD
         if self._krn_weights is not None:
-            size += len(self._krn_weights) * BYTES_PER_WORD
+            size += sum(len(x) for x in self._krn_weights) * BYTES_PER_WORD
         if self._krn_delays is not None:
-            size += len(self._krn_delays) * BYTES_PER_WORD
+            size += sum(len(x) for x in self._krn_delays) * BYTES_PER_WORD
         return size
 
     @overrides(AbstractGenerateConnectorOnMachine.get_connected_vertices)