Skip to content

Commit

Permalink
Fix type
Browse files Browse the repository at this point in the history
  • Loading branch information
rowleya committed Jun 20, 2024
1 parent 4ebaaad commit 3fe23a0
Showing 1 changed file with 4 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 3fe23a0

Please sign in to comment.