From 81b4fee904b2aedf3c2da87cd795960ba33f9cc7 Mon Sep 17 00:00:00 2001
From: drothschedl <dominik.rothschedl@dwh.at>
Date: Mon, 25 Jul 2022 11:19:52 +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 c205ca6..b05ce0b 100644
--- a/Environment.py
+++ b/Environment.py
@@ -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 = []