Updating documentation format.

1 files changed, 5 insertions, 5 deletions

diff --git a/src/argaze/GazeFeatures.py b/src/argaze/GazeFeatures.py
index 6209dc9..0e40238 100644
--- a/src/argaze/GazeFeatures.py
+++ b/src/argaze/GazeFeatures.py
@@ -88,7 +88,7 @@ class GazePosition():
         else:
             return distance < self.precision
 
-    def draw(self, frame, color=(0, 255, 255), draw_precision=True):
+    def draw(self, frame: numpy.array, color=(0, 255, 255), draw_precision=True):
         """Draw gaze position point and precision circle."""
 
         if self.valid:
@@ -190,7 +190,7 @@ class GazeMovement():
 
         return output
 
-    def draw_positions(self, frame, color=(0, 55, 55)):
+    def draw_positions(self, frame: numpy.array, color=(0, 55, 55)):
         """Draw gaze movement positions"""
 
         gaze_positions = self.positions.copy()
@@ -224,7 +224,7 @@ class Fixation(GazeMovement):
 
         raise NotImplementedError('merge() method not implemented')
 
-    def draw(self, frame, color):
+    def draw(self, frame: numpy.array, color):
         """Draw fixation into frame."""
 
         raise NotImplementedError('draw() method not implemented')
@@ -241,7 +241,7 @@ class Saccade(GazeMovement):
 
         super().__post_init__()
 
-    def draw(self, frame, color):
+    def draw(self, frame: numpy.array, color):
         """Draw saccade into frame."""
 
         raise NotImplementedError('draw() method not implemented')
@@ -442,7 +442,7 @@ class ScanPath(list):
 
         self.__last_fixation = fixation
 
-    def draw(self, frame, fixation_color=(255, 255, 255), saccade_color=(255, 255, 255), deepness=0):
+    def draw(self, frame: numpy.array, fixation_color=(255, 255, 255), saccade_color=(255, 255, 255), deepness=0):
         """Draw scan path into frame."""
 
         last_step = None