Addresses minor pylint issues

Co-authored-by: Amy Jordan <amy_jordan@live.co.uk> Co-authored-by: Kim Diep <kim.diep.techno@gmail.com>
jpalmerr · May 21, 2019 · 20972fd · 20972fd
1 parent ac2154d
commit 20972fd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lib/drawing.py b/lib/drawing.py
@@ -7,9 +7,9 @@ def __init__(self, image):
         self.image = image
 
     def reshape(self):
-        img = (self.image).resize((28,28)).convert('L')
+        img = (self.image).resize((28, 28)).convert('L')
         np_img = np.array(img)
         np_img = np.expand_dims(np_img, axis=0)
-        np_img = np.reshape(np_img, (28,28,1))
+        np_img = np.reshape(np_img, (28, 28, 1))
         np_img = normalize(np_img)
         return np_img
diff --git a/lib/prediction.py b/lib/prediction.py
@@ -6,7 +6,7 @@
 
 class Prediction():
     def __init__(self, drawing):
-        self.drawing= drawing
+        self.drawing = drawing
 
     def predict(self):
         model = load_model("./models/first_model_training.h5")