Update these calls too

SpiNNakerManchester · Oct 10, 2024 · a136d26 · a136d26
1 parent 6a19175
commit a136d26
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 7 deletions.
diff --git a/gfe_examples/Conways/partitioned_example_a_no_vis_no_buffer/conways_basic_cell.py b/gfe_examples/Conways/partitioned_example_a_no_vis_no_buffer/conways_basic_cell.py
@@ -104,8 +104,8 @@ def generate_machine_data_specification(
             label="results")
 
         # write key needed to transmit with
-        key = FecDataView.get_routing_infos().get_first_key_from_pre_vertex(
-            self, self.PARTITION_ID)
+        r_infos = FecDataView.get_routing_infos()
+        key = r_infos.get_single_first_key_from_pre_vertex(self)
 
         spec.switch_write_focus(DataRegions.TRANSMISSIONS)
         spec.write_value(int(key is not None))

diff --git a/gfe_examples/Conways/partitioned_example_b_no_vis_buffer/conways_basic_cell.py b/gfe_examples/Conways/partitioned_example_b_no_vis_buffer/conways_basic_cell.py
@@ -113,8 +113,8 @@ def generate_machine_data_specification(
              FecDataView.get_max_run_time_steps()])
 
         # write key needed to transmit with
-        key = FecDataView.get_routing_infos().get_first_key_from_pre_vertex(
-            self, self.PARTITION_ID)
+        r_infos = FecDataView.get_routing_infos()
+        key = r_infos.get_single_first_key_from_pre_vertex(self)
 
         spec.switch_write_focus(DataRegions.TRANSMISSIONS)
         spec.write_value(int(key is not None))

diff --git a/gfe_examples/sync_test/sync_test_vertex.py b/gfe_examples/sync_test/sync_test_vertex.py
@@ -82,9 +82,8 @@ def generate_data_specification(
             spec.write_value(0)
         else:
             routing_info = FecDataView.get_routing_infos()
-            key = routing_info.get_first_key_from_pre_vertex(
+            key = routing_info.get_safe_first_key_from_pre_vertex(
                 self, SEND_PARTITION)
-            assert key is not None
             spec.write_value(key)
 
         # End-of-Spec:

diff --git a/gfe_examples/template/template_vertex.py b/gfe_examples/template/template_vertex.py
@@ -111,7 +111,7 @@ def _reserve_app_memory_regions(self, spec):
     def _write_app_memory_regions(self, spec, iptags):
         # Get the key, assuming all outgoing edges use the same key
         routing_info = FecDataView.get_routing_infos()
-        key = routing_info.get_first_key_from_pre_vertex(self, PARTITION_ID)
+        key = routing_info.get_single_first_key_from_pre_vertex(self)
 
         # Write the transmission region
         spec.switch_write_focus(DataRegions.TRANSMISSION)