Formats code to flake8

Author: rbaron

.dockerignore

@@ -0,0 +1 @@
+venv/

.flake8

@@ -0,0 +1,2 @@
+[flake8]
+max-line-length=100

Dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.7
+
+WORKDIR /opt/omr
+
+COPY . .
+
+RUN pip3 install -r requirements.txt
+
+CMD ["python", "omr.py", "--help"]

omr.py

@@ -1,6 +1,5 @@
 import argparse
 import cv2
-import math
 import numpy as np
 
 # When the four corners are identified, we will do a four-point
@@ -24,6 +23,7 @@
 ALTERNATIVE_WIDTH = 50
 ALTERNATIVE_WIDTH_WITH_MARGIN = 100
 
+
 def calculate_contour_features(contour):
     """Calculates interesting properties (features) of a contour.
 
@@ -73,8 +73,9 @@ def calculate_corner_features():
     # contour (that is, it is _not_ the outer contour) to be the corner contour.
     # If in trouble, verify that this contour is the corner contour with
     # cv2.drawContours(corner_img, [corner_contour], -1, (255, 0, 0))
-    corner_contour = next(ct for i, ct in enumerate(contours)
-                             if hierarchy[0][i][3] != -1)
+    corner_contour = next(ct
+                          for i, ct in enumerate(contours)
+                          if hierarchy[0][i][3] != -1)
 
     return calculate_contour_features(corner_contour)
 
@@ -185,7 +186,6 @@ def perspective_transform(img, points):
         [TRANSF_SIZE, 0]],
         dtype="float32")
 
-    img_dest = img.copy()
     transf = cv2.getPerspectiveTransform(source, dest)
     warped = cv2.warpPerspective(img, transf, (TRANSF_SIZE, TRANSF_SIZE))
     return warped
@@ -210,8 +210,8 @@ def get_question_patch(transf, question_index):
     br = sheet_coord_to_transf_coord(
         ANSWER_SHEET_WIDTH - ANSWER_PATCH_RIGHT_MARGIN,
         FIRST_ANSWER_PATCH_TOP_Y +
-            ANSWER_PATCH_HEIGHT +
-            ANSWER_PATCH_HEIGHT_WITH_MARGIN * question_index
+        ANSWER_PATCH_HEIGHT +
+        ANSWER_PATCH_HEIGHT_WITH_MARGIN * question_index
     )
     return transf[tl[1]:br[1], tl[0]:br[0]]
 
@@ -224,7 +224,8 @@ def get_question_patches(transf):
 def get_alternative_patches(question_patch):
     for i in range(5):
         x0, _ = sheet_coord_to_transf_coord(ALTERNATIVE_WIDTH_WITH_MARGIN * i, 0)
-        x1, _ = sheet_coord_to_transf_coord(ALTERNATIVE_WIDTH + ALTERNATIVE_WIDTH_WITH_MARGIN * i, 0)
+        x1, _ = sheet_coord_to_transf_coord(ALTERNATIVE_WIDTH +
+                                            ALTERNATIVE_WIDTH_WITH_MARGIN * i, 0)
         yield question_patch[:, x0:x1]
 
 
@@ -270,9 +271,6 @@ def get_answers(source_file):
     - Apply perpsective transform to get a bird's eye view
     - Scan each line for the marked alternative
     """
-
-    corner_features = calculate_corner_features()
-
     im_orig = cv2.imread(source_file)
 
     im_normalized = normalize(im_orig)
@@ -338,4 +336,3 @@ def main():
 
 if __name__ == '__main__':
     main()
-

test_omr.py

@@ -10,4 +10,3 @@ def test_recognizes_digitally_marked_answer_sheet():
 def test_recognizes_photo_of_digitally_marked_answer_sheet():
     answers = get_answers(os.path.join("img", "answered-sheet-photo.jpg"))
     assert answers[0] == ['A', 'C', 'C', 'E', 'N/A', 'N/A', 'A', 'N/A', 'N/A', 'N/A']
-