From b64209e994ed36edb49f0a6b3da68f0429df062b Mon Sep 17 00:00:00 2001 From: "Christian Y. Brenninkmeijer" Date: Mon, 24 Jun 2024 08:05:55 +0100 Subject: [PATCH] type fixes --- .../test_multi_src_internal/sdram_splitter.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/gfe_integration_tests/sdram_edge_tests/test_multi_src_internal/sdram_splitter.py b/gfe_integration_tests/sdram_edge_tests/test_multi_src_internal/sdram_splitter.py index c743365d..f025dad0 100644 --- a/gfe_integration_tests/sdram_edge_tests/test_multi_src_internal/sdram_splitter.py +++ b/gfe_integration_tests/sdram_edge_tests/test_multi_src_internal/sdram_splitter.py @@ -34,7 +34,7 @@ class SDRAMSplitter(AbstractSplitterCommon): def __init__(self) -> None: super().__init__() self.__post_vertex = None - self._pre_vertices: List[SourceSegmentedSDRAMMachinePartition] = list() + self._pre_vertices: List[SDRAMMachineVertex] = list() @property def _post_vertex(self): @@ -44,6 +44,7 @@ def _post_vertex(self): @overrides(AbstractSplitterCommon.get_out_going_vertices) def get_out_going_vertices( self, partition_id: str) -> List[MachineVertex]: + assert isinstance(self.__post_vertex, SDRAMMachineVertex) return [self.__post_vertex] @overrides(AbstractSplitterCommon.get_in_coming_vertices) @@ -88,7 +89,7 @@ def create_machine_vertices(self, chip_counter): chip_counter.add_core(pre_vertex.sdram_required) @overrides(AbstractSplitterCommon.get_out_going_slices) - def get_out_going_slices(self) -> Slice: + def get_out_going_slices(self) -> List[Slice]: return [self._post_vertex.vertex_slice] @overrides(AbstractSplitterCommon.get_in_coming_slices)