diff --git a/gfe_integration_tests/sdram_edge_tests/test_constant_internal/sdram_splitter_internal.py b/gfe_integration_tests/sdram_edge_tests/test_constant_internal/sdram_splitter_internal.py index 4fac689a..1a0b92e8 100644 --- a/gfe_integration_tests/sdram_edge_tests/test_constant_internal/sdram_splitter_internal.py +++ b/gfe_integration_tests/sdram_edge_tests/test_constant_internal/sdram_splitter_internal.py @@ -47,11 +47,13 @@ def _post_vertex(self): return self.__post_vertex @overrides(AbstractSplitterCommon.get_out_going_vertices) - def get_out_going_vertices(self, partition_id: str) -> List[SDRAMMachineVertex]: + def get_out_going_vertices(self, partition_id: str) -> List[ + SDRAMMachineVertex]: return [self._pre_vertex] @overrides(AbstractSplitterCommon.get_in_coming_vertices) - def get_in_coming_vertices(self, partition_id: str) -> list[SDRAMMachineVertex]: + def get_in_coming_vertices(self, partition_id: str) -> List[ + SDRAMMachineVertex]: return [self._post_vertex] def create_machine_vertices(self, chip_counter):