From 23ac63d55b2021bf1053fcd33b5e98482079455f Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Tue, 14 Jan 2025 12:48:43 +0100
Subject: [PATCH] Fix more typing

---
 face_swapper/src/generator.py | 4 +++-
 face_swapper/src/helper.py    | 3 ++-
 face_swapper/src/training.py  | 3 ++-
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/face_swapper/src/generator.py b/face_swapper/src/generator.py
index e93a668..2f92bc3 100644
--- a/face_swapper/src/generator.py
+++ b/face_swapper/src/generator.py
@@ -1,7 +1,9 @@
+from typing import Tuple
+
 import torch
 import torch.nn as nn
 
-from .typing import IDEmbedding, TargetAttributes, Tensor, Tuple
+from .typing import IDEmbedding, TargetAttributes, Tensor
 
 
 class AdaptiveEmbeddingIntegrationNetwork(nn.Module):
diff --git a/face_swapper/src/helper.py b/face_swapper/src/helper.py
index 6f42cd2..954e0b6 100644
--- a/face_swapper/src/helper.py
+++ b/face_swapper/src/helper.py
@@ -1,8 +1,9 @@
 import configparser
+from typing import Tuple
 
 import torch
 
-from .typing import Tensor, Tuple
+from .typing import Tensor
 
 CONFIG = configparser.ConfigParser()
 CONFIG.read('config.ini')
diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py
index 12d740b..6070d3c 100644
--- a/face_swapper/src/training.py
+++ b/face_swapper/src/training.py
@@ -1,5 +1,6 @@
 import configparser
 import os
+from typing import Tuple
 
 import cv2
 import numpy
@@ -17,7 +18,7 @@ from .data_loader import DataLoaderVGG, read_image
 from .discriminator import MultiscaleDiscriminator
 from .generator import AdaptiveEmbeddingIntegrationNetwork
 from .helper import L2_loss, hinge_loss
-from .typing import Batch, DiscriminatorOutputs, IDEmbedding, LossDict, TargetAttributes, Tensor, Tuple
+from .typing import Batch, DiscriminatorOutputs, IDEmbedding, LossDict, TargetAttributes, Tensor
 
 CONFIG = configparser.ConfigParser()
 CONFIG.read('config.ini')