Flake8 fix

.gitignore

@@ -1 +1,2 @@
-__pycache__
+__pycache__
+.vscode

video_heatmap.py

@@ -22,6 +22,7 @@
 import vision
 import arguments_parser
 
+
 def main():
     """
     Whole heatmap pipeline creation.
@@ -50,7 +51,7 @@ def main():
 
                 erodated_image = vision.apply_morph(background_filter,
                                                     morph_type=cv2.MORPH_ERODE,
-                                                    kernel_size=(5,5))
+                                                    kernel_size=(5, 5))
                 accumulated_image = vision.add_images(accumulated_image, erodated_image)
                 normalized_image = vision.normalize_image(accumulated_image)
                 heatmap_image = vision.apply_heatmap_colors(normalized_image)
@@ -70,5 +71,6 @@ def main():
     capture.release()
     cv2.destroyAllWindows()
 
+
 if __name__ == '__main__':
     main()

vision.py

@@ -10,7 +10,7 @@
 
 def apply_morph(image: np.ndarray,
                 morph_type=cv2.MORPH_CLOSE,
-                kernel_size: Tuple[int, int] = (3,3),
+                kernel_size: Tuple[int, int] = (3, 3),
                 make_gaussian: bool = True):
     """
     Apply opencv morphological operation to image and return it.
@@ -27,11 +27,11 @@ def apply_morph(image: np.ndarray,
     """
     kernel = cv2.getStructuringElement(cv2.MORPH_RECT, kernel_size)
     if make_gaussian:
-        image = cv2.GaussianBlur(image,(3,3),0)
+        image = cv2.GaussianBlur(image, (3, 3), 0)
     return cv2.morphologyEx(image, morph_type, kernel)
 
 
-def add_images(image1: np.ndarray, 
+def add_images(image1: np.ndarray,
                image2: np.ndarray) -> np.ndarray:
     """
     Add two images together. Colors values can be bigger then 255 restriction.
@@ -88,4 +88,3 @@ def superimpose(image1: np.ndarray,
     """
 
     return cv2.addWeighted(image1, alpha, image2, 1 - alpha, 0.0)
-