Skip to content

Commit 1fe5e38

Browse files
authored
Merge pull request #55 from harisankar95/main
updated return type annotation
2 parents a76910a + 71d41bb commit 1fe5e38

1 file changed

Lines changed: 4 additions & 8 deletions

File tree

pathfinding/core/world.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from typing import Dict
1+
from typing import Dict, List
22

33
from .diagonal_movement import DiagonalMovement
44
from .grid import Grid
@@ -10,14 +10,10 @@ class World:
1010
def __init__(self, grids: Dict[int, Grid]):
1111
self.grids = grids
1212

13-
def neighbors(
14-
self, node: Node, diagonal_movement: DiagonalMovement
15-
) -> list[Node]:
16-
return self.grids[node.grid_id].neighbors(
17-
node, diagonal_movement=diagonal_movement)
13+
def neighbors(self, node: Node, diagonal_movement: DiagonalMovement) -> List[Node]:
14+
return self.grids[node.grid_id].neighbors(node, diagonal_movement=diagonal_movement)
1815

1916
def calc_cost(self, node_a, node_b, weighted=False):
2017
# TODO: if node_a.grid_id != node_b.grid_id calculate distance between
2118
# grids as well, for now we ignore switching grids
22-
return self.grids[node_a.grid_id].calc_cost(
23-
node_a, node_b, weighted=weighted)
19+
return self.grids[node_a.grid_id].calc_cost(node_a, node_b, weighted=weighted)

0 commit comments

Comments
 (0)