Use mypy and flake8 for face swapper

.github/workflows/ci.yml

@@ -15,5 +15,5 @@ jobs:
   - run: pip install flake8
   - run: pip install flake8-import-order
   - run: pip install mypy
-  - run: flake8 arcface_converter
-  - run: mypy arcface_converter
+  - run: flake8 arcface_converter face_swapper
+  - run: mypy arcface_converter face_swapper

face_swapper/src/discriminator.py

@@ -1,7 +1,9 @@
+from typing import List
+
 import numpy
 import torch.nn as nn
 
-from .typing import DiscriminatorOutputs, List, Tensor
+from .typing import DiscriminatorOutputs, Tensor
 
 
 class NLayerDiscriminator(nn.Module):

face_swapper/train.py

@@ -2,6 +2,5 @@
 
 from src.training import train
 
-
 if __name__ == '__main__':
 	train()