From 21d4461ba99ffd5082b4bc0129657f9d5f4ef4c2 Mon Sep 17 00:00:00 2001
From: drothschedl <dominik.rothschedl@dwh.at>
Date: Mon, 25 Jul 2022 11:21:08 +0200
Subject: [PATCH] add type to parameter

---
 Environment.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Environment.py b/Environment.py
index 8722e3f..0208911 100644
--- a/Environment.py
+++ b/Environment.py
@@ -84,7 +84,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:
@@ -103,7 +103,7 @@ class Env:
         Plot graph
     '''
     
-    def plot_graph(self, path):
+    def plot_graph(self, path: list): -> None
         node_color = []
         edge_color = []
         edge_width = []