mypy

architecture.py

@@ -4,10 +4,10 @@
 
 # third-party imports
 import tensorflow as tf
-from tensorflow.keras import Model  # type: ignore
-from tensorflow.keras.losses import Loss  # type: ignore
-from tensorflow.keras.optimizers import Optimizer  # type: ignore
-from PIL import Image  # type: ignore
+from tensorflow.keras import Model
+from tensorflow.keras.losses import Loss
+from tensorflow.keras.optimizers import Optimizer
+from PIL import Image
 
 # module imports
 from transforms import ImageTransform

create_data_lists.py

@@ -4,7 +4,7 @@
 from typing import List
 
 # third-party imports
-from PIL import Image  # type: ignore
+from PIL import Image
 
 
 def create_data_lists(

evaluator.py

@@ -4,9 +4,9 @@
 
 # third-party imports
 import tensorflow as tf
-from tensorflow.keras import Model  # type: ignore
-from tensorflow.keras.metrics import Mean  # type: ignore
-from PIL import Image, ImageDraw, ImageFont  # type: ignore
+from tensorflow.keras import Model
+from tensorflow.keras.metrics import Mean
+from PIL import Image, ImageDraw, ImageFont
 from skimage.metrics import peak_signal_noise_ratio, structural_similarity
 
 # module imports

model.py

@@ -3,9 +3,9 @@
 from typing import List, Optional
 
 # third-party imports
-import tensorflow as tf  # type: ignore
-from tensorflow.keras import layers, Model  # type: ignore
-from tensorflow.keras.applications import VGG19  # type: ignore
+import tensorflow as tf
+from tensorflow.keras import layers, Model
+from tensorflow.keras.applications import VGG19
 
 
 class ConvolutionalBlock(layers.Layer):

mypy.ini

@@ -0,0 +1,2 @@
+[mypy]
+ignore_missing_imports = True

resolve.py

@@ -1,5 +1,5 @@
 # third-party imports
-import tensorflow as tf  # type: ignore
+import tensorflow as tf
 from pprint import PrettyPrinter
 
 # module imports

test/conftest.py

@@ -2,8 +2,8 @@
 import os
 
 # third-party imports
-import pytest  # type: ignore
-import tensorflow as tf  # type: ignore
+import pytest
+import tensorflow as tf
 
 
 @pytest.fixture(params=["CPU", "GPU"], autouse=True)

test/test_dataset.py

@@ -3,9 +3,9 @@
 from unittest.mock import patch, MagicMock
 
 # third party imports
-import pytest  # type: ignore
-import tensorflow as tf  # type: ignore
-from PIL import Image  # type: ignore
+import pytest
+import tensorflow as tf
+from PIL import Image
 
 # module imports
 from trainer import Trainer

test/test_model.py

@@ -2,8 +2,8 @@
 import unittest.mock as mock
 
 # third-party imports
-import pytest  # type: ignore
-import tensorflow as tf  # type: ignore
+import pytest
+import tensorflow as tf
 
 # module imports
 from model import (

test/test_transforms.py

@@ -1,8 +1,8 @@
 # third-party imports
-import pytest  # type: ignore
+import pytest
 import numpy as np
-import tensorflow as tf  # type: ignore
-from PIL import Image  # type: ignore
+import tensorflow as tf
+from PIL import Image
 
 # module imports
 from transforms import ImageTransform

trainer.py

@@ -5,9 +5,9 @@
 import logging
 
 # third-party imports
-import tensorflow as tf  # type: ignore
-from colorama import Fore, init  # type: ignore
-from PIL import Image  # type: ignore
+import tensorflow as tf
+from colorama import Fore, init
+from PIL import Image
 
 # module imports
 from transforms import ImageTransform

transforms.py

@@ -4,8 +4,8 @@
 
 # third-party imports
 import numpy as np
-import tensorflow as tf  # type: ignore
-from PIL import Image  # type: ignore
+import tensorflow as tf
+from PIL import Image
 
 
 class ImageTransform(object):