From 85209e889c34b3c2ffb9d6a9436c3ac2b84af055 Mon Sep 17 00:00:00 2001 From: "Christian Y. Brenninkmeijer" Date: Mon, 8 Jan 2024 10:11:08 +0000 Subject: [PATCH] typing --- spinnaker_graph_front_end/utilities/data_utils.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/spinnaker_graph_front_end/utilities/data_utils.py b/spinnaker_graph_front_end/utilities/data_utils.py index c064e514..b18c27bc 100644 --- a/spinnaker_graph_front_end/utilities/data_utils.py +++ b/spinnaker_graph_front_end/utilities/data_utils.py @@ -12,12 +12,15 @@ # See the License for the specific language governing permissions and # limitations under the License. +from pacman.model.graphs.machine import MachineVertex from spinn_front_end_common.utilities.constants import SIMULATION_N_BYTES +from spinn_front_end_common.interface.ds import DataSpecificationGenerator from spinn_front_end_common.interface.simulation.simulation_utilities import ( get_simulation_header_array, get_simulation_header_array_no_timestep) -def generate_system_data_region(spec, region_id, machine_vertex): +def generate_system_data_region(spec: DataSpecificationGenerator, + region_id: int, machine_vertex: MachineVertex): """ Generate a system data region for time-based simulations. @@ -38,7 +41,9 @@ def generate_system_data_region(spec, region_id, machine_vertex): machine_vertex.get_binary_file_name())) -def generate_steps_system_data_region(spec, region_id, machine_vertex): +def generate_steps_system_data_region( + spec: DataSpecificationGenerator, region_id: int, + machine_vertex: MachineVertex): """ Generate a system data region for step-based simulations.