asim-rl
/
graph
2
0
Fork 0
Code Releases Activity

add type to parameter

This commit is contained in:
drothschedl 2022-07-25 11:19:52 +02:00
parent 21bcbf49c1
commit 81b4fee904
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Environment.py
View File

@ -83,7 +83,7 @@ class Env:
'''
Get shortest calcualted path and print it
'''
def print_shortest_path(self, Q):
def print_shortest_path(self, Q: dict): -> None
path = [self.start]
state = self.start
while state!=self.goal:
@ -101,7 +101,7 @@ class Env:
'''
Plot graph
'''
def plot_graph(self, path):
def plot_graph(self, path: list): -> None
node_color = []
edge_color = []
edge_width = []