fix: added typing

1 files changed, 6 insertions, 4 deletions

diff --git a/Trail.py b/Trail.py
index 08450d3..5dd024d 100644
--- a/Trail.py
+++ b/Trail.py
@@ -1,10 +1,12 @@
 
+from typing import Union
+
 from Vec import Vec2
-import Trail
+from Door import Door
 
 class Trail:
 
-    def __init__(self, pos: Vec2, nextTrail: Trail):
+    def __init__(self, pos: Vec2, nextTrail: Union['Trail', 'Door']):
         self._isOn = False
         self.pos = pos
         self._nextTrail = nextTrail
@@ -14,11 +16,11 @@ class Trail:
         return self._isOn
 
 
-    def activate(self):
+    def activate(self) -> None:
         self._isOn = True
         self._nextTrail.activate()
 
 
-    def deactivate(self):
+    def deactivate(self) -> None:
         self._isOn = False
         self._nextTrail.deactivate()