Fix CI

face_swapper/src/models/generator.py

@@ -5,7 +5,7 @@ from torch import Tensor, nn
 
 from ..networks.aad import AAD
 from ..networks.unet import UNet
-from ..types import Feature, Embedding
+from ..types import Embedding, Feature
 
 
 class Generator(nn.Module):

face_swapper/src/models/loss.py

@@ -7,7 +7,7 @@ from torch import Tensor, nn
 from torchvision import transforms
 
 from ..helper import calc_embedding
-from ..types import Feature, EmbedderModule, FaceParserModule, GazerModule, Loss, Mask, MotionExtractorModule
+from ..types import EmbedderModule, FaceParserModule, Feature, GazerModule, Loss, Mask, MotionExtractorModule
 
 
 class DiscriminatorLoss(nn.Module):

face_swapper/src/networks/aad.py

@@ -4,7 +4,7 @@ from typing import Tuple
 import torch
 from torch import Tensor, nn
 
-from ..types import Feature, Embedding
+from ..types import Embedding, Feature
 
 
 class AAD(nn.Module):