Fix ReverseIPTagMulticastSource usage

SpiNNakerManchester · Oct 10, 2024 · 8ea75c8 · 8ea75c8
1 parent 730c2c0
commit 8ea75c8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 10 deletions.
diff --git a/spynnaker/pyNN/models/spike_source/spike_source_array_vertex.py b/spynnaker/pyNN/models/spike_source/spike_source_array_vertex.py
@@ -115,8 +115,7 @@ def __init__(
             max_atoms_per_core: Union[int, Tuple[int, ...]],
             model: SpikeSourceArray,
             splitter: Optional[AbstractSplitterCommon],
-            n_colour_bits: Optional[int],
-            partition_id: str):
+            n_colour_bits: Optional[int]):
         # pylint: disable=too-many-arguments
         self.__model_name = "SpikeSourceArray"
         self.__model = model
@@ -134,7 +133,6 @@ def __init__(
             n_keys=n_neurons, label=label,
             max_atoms_per_core=max_atoms_per_core,
             send_buffer_times=_send_buffer_times(spike_times, time_step),
-            send_buffer_partition_id=partition_id,
             splitter=splitter)
 
         self._check_spike_density(spike_times)

diff --git a/spynnaker/pyNN/models/utility_models/spike_injector/spike_injector_vertex.py b/spynnaker/pyNN/models/utility_models/spike_injector/spike_injector_vertex.py
@@ -45,8 +45,7 @@ class SpikeInjectorVertex(
         "__structure")
 
     default_parameters = {
-        'label': "spikeInjector", 'port': None, 'virtual_key': None,
-        'partition_id': SPIKE_PARTITION_ID}
+        'label': "spikeInjector", 'port': None, 'virtual_key': None}
 
     SPIKE_RECORDING_REGION_ID = 0
 
@@ -56,14 +55,12 @@ def __init__(
             reserve_reverse_ip_tag: bool,
             splitter: Optional[AbstractSplitterCommon],
             max_atoms_per_core: Optional[
-                Union[int, Tuple[int, ...]]] = sys.maxsize,
-            partition_id: str = SPIKE_PARTITION_ID):
+                Union[int, Tuple[int, ...]]] = sys.maxsize):
         # pylint: disable=too-many-arguments
         super().__init__(
             n_keys=n_neurons, label=label, receive_port=port,
             virtual_key=virtual_key,
             reserve_reverse_ip_tag=reserve_reverse_ip_tag,
-            injection_partition_id=partition_id,
             splitter=splitter, max_atoms_per_core=max_atoms_per_core)
 
         # Set up for recording

diff --git a/spynnaker/pyNN/spynnaker_external_device_plugin_manager.py b/spynnaker/pyNN/spynnaker_external_device_plugin_manager.py
@@ -272,8 +272,7 @@ def add_poisson_live_rate_control(
         controller = ReverseIpTagMultiCastSource(
             n_keys=vertex.n_atoms, label=control_label,
             receive_port=receive_port,
-            reserve_reverse_ip_tag=reserve_reverse_ip_tag,
-            injection_partition_id=LIVE_POISSON_CONTROL_PARTITION_ID)
+            reserve_reverse_ip_tag=reserve_reverse_ip_tag)
         SpynnakerExternalDevicePluginManager.add_application_vertex(controller)
         edge = SpynnakerExternalDevicePluginManager.add_edge(
             controller, vertex, LIVE_POISSON_CONTROL_PARTITION_ID)