Type fixes

SpiNNakerManchester · Oct 4, 2024 · 3cdf429 · 3cdf429
1 parent 11efb06
commit 3cdf429
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/spynnaker/pyNN/extra_algorithms/delay_support_adder.py b/spynnaker/pyNN/extra_algorithms/delay_support_adder.py
@@ -28,14 +28,14 @@
 
 
 def delay_support_adder() -> Tuple[
-        Sequence[DelayExtensionVertex], Sequence[ApplicationEdge]]:
+        Sequence[DelayExtensionVertex], Sequence[Tuple[ApplicationEdge, str]]]:
     """
     Adds the delay extensions to the application graph, now that all the
     splitter objects have been set.
 
     :return: The delay vertices and delay edges that were added
     :rtype: tuple(list(DelayExtensionVertex), list(DelayedApplicationEdge or
-        DelayAfferentApplicationEdge))
+        DelayAfferentApplicationEdge, str))
     """
     adder = _DelaySupportAdder()
     # pylint: disable=protected-access

diff --git a/spynnaker/pyNN/models/neuron/local_only/local_only_convolution.py b/spynnaker/pyNN/models/neuron/local_only/local_only_convolution.py
@@ -78,7 +78,7 @@ def __init__(self, delay: Weight_Delay_In_Types = None):
 
         # Store the sources to avoid recalculation
         self.__cached_sources: Dict[ApplicationVertex, Dict[
-                Tuple(PopulationApplicationVertex, str),
+                Tuple[PopulationApplicationVertex, str],
                 List[Source]]] = dict()
 
     @property
@@ -250,7 +250,7 @@ def write_parameters(
 
     def __get_sources_for_target(
             self, app_vertex: AbstractPopulationVertex) -> Dict[
-                Tuple(PopulationApplicationVertex, str), List[Source]]:
+                Tuple[PopulationApplicationVertex, str], List[Source]]:
         """
         Get all the application vertex sources that will hit the given
         application vertex.

diff --git a/spynnaker/pyNN/models/neuron/local_only/local_only_pool_dense.py b/spynnaker/pyNN/models/neuron/local_only/local_only_pool_dense.py
@@ -74,7 +74,7 @@ def __init__(self, delay: Weight_Delay_In_Types = None):
         """
         # Store the sources to avoid recalculation
         self.__cached_sources: Dict[ApplicationVertex, Dict[
-                Tuple(ApplicationVertex, str), List[Source]]] = dict()
+                Tuple[ApplicationVertex, str], List[Source]]] = dict()
 
         super().__init__(delay)
         if not isinstance(self.delay, (float, int)):