Use mypy and flake8 for face swapper

facefusion · Jan 14, 2025 · d00c0e2 · d00c0e2
1 parent 13f29eb
commit d00c0e2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/.github/workflows/ci.yml b/.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
diff --git a/face_swapper/src/discriminator.py b/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):

diff --git a/face_swapper/train.py b/face_swapper/train.py
@@ -2,6 +2,5 @@
 
 from src.training import train
 
-
 if __name__ == '__main__':
 	train()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,6 +2,5 @@

		from src.training import train


		if __name__ == '__main__':
		train()