Rename functions

SpiNNakerManchester · Oct 14, 2024 · 42b9b38 · 42b9b38
1 parent 4fc9f33
commit 42b9b38
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/spinn_gym/games/breakout/breakout_machine_vertex.py b/spinn_gym/games/breakout/breakout_machine_vertex.py
@@ -150,7 +150,7 @@ def generate_data_specification(
         spec.switch_write_focus(
             BreakoutMachineVertex._BREAKOUT_REGIONS.BREAKOUT.value)
         routing_info = SpynnakerDataView.get_routing_infos()
-        spec.write_value(routing_info.get_safe_first_key_from_pre_vertex(
+        spec.write_value(routing_info.get_key_from(
             vertex, SPIKE_PARTITION_ID))
         if self.app_vertex.source_vertex is None:
             raise ValueError(

diff --git a/spinn_gym/games/double_inverted_pendulum/double_pendulum_machine_vertex.py b/spinn_gym/games/double_inverted_pendulum/double_pendulum_machine_vertex.py
@@ -167,7 +167,7 @@ def generate_data_specification(
         spec.switch_write_focus(
             self._DOUBLE_PENDULUM_REGIONS.PENDULUM.value)
         routing_info = SpynnakerDataView.get_routing_infos()
-        spec.write_value(routing_info.get_safe_first_key_from_pre_vertex(
+        spec.write_value(routing_info.get_key_from(
             vertex, SPIKE_PARTITION_ID))
 
         # Write recording region for score

diff --git a/spinn_gym/games/inverted_pendulum/inverted_pendulum_machine_vertex.py b/spinn_gym/games/inverted_pendulum/inverted_pendulum_machine_vertex.py
@@ -165,7 +165,7 @@ def generate_data_specification(
         spec.switch_write_focus(
             self._PENDULUM_REGIONS.PENDULUM.value)
         routing_info = SpynnakerDataView.get_routing_infos()
-        spec.write_value(routing_info.get_safe_first_key_from_pre_vertex(
+        spec.write_value(routing_info.get_key_from(
             vertex, SPIKE_PARTITION_ID))
 
         # Write recording region for score

diff --git a/spinn_gym/games/logic/logic_machine_vertex.py b/spinn_gym/games/logic/logic_machine_vertex.py
@@ -150,7 +150,7 @@ def generate_data_specification(
         spec.switch_write_focus(
             self._LOGIC_REGIONS.LOGIC.value)
         routing_info = SpynnakerDataView.get_routing_infos()
-        spec.write_value(routing_info.get_safe_first_key_from_pre_vertex(
+        spec.write_value(routing_info.get_key_from(
             vertex, SPIKE_PARTITION_ID))
 
         # Write recording region for score

diff --git a/spinn_gym/games/multi_arm_bandit/bandit_machine_vertex.py b/spinn_gym/games/multi_arm_bandit/bandit_machine_vertex.py
@@ -160,7 +160,7 @@ def generate_data_specification(
         spec.switch_write_focus(
             self._BANDIT_REGIONS.BANDIT.value)
         routing_info = SpynnakerDataView.get_routing_infos()
-        spec.write_value(routing_info.get_safe_first_key_from_pre_vertex(
+        spec.write_value(routing_info.get_key_from(
             vertex, SPIKE_PARTITION_ID))
 
         # Write recording region for score

diff --git a/spinn_gym/games/store_recall/store_recall_machine_vertex.py b/spinn_gym/games/store_recall/store_recall_machine_vertex.py
@@ -154,7 +154,7 @@ def generate_data_specification(
         spec.switch_write_focus(
             self._RECALL_REGIONS.RECALL.value)
         routing_info = SpynnakerDataView.get_routing_infos()
-        spec.write_value(routing_info.get_safe_first_key_from_pre_vertex(
+        spec.write_value(routing_info.get_key_from(
             vertex, SPIKE_PARTITION_ID))
 
         # Write recording region for score