diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py
index 637f094..df97894 100644
--- a/face_swapper/src/training.py
+++ b/face_swapper/src/training.py
@@ -20,7 +20,7 @@ from .models.generator import Generator
 from .models.loss import AdversarialLoss, AttributeLoss, DiscriminatorLoss, GazeLoss, IdentityLoss, PoseLoss, ReconstructionLoss
 from .types import Batch, Embedding, OptimizerConfig, WarpTemplate
 
-warnings.filterwarnings("ignore", category=UserWarning, module="torch")
+warnings.filterwarnings('ignore', category = UserWarning, module = 'torch')
 
 CONFIG = configparser.ConfigParser()
 CONFIG.read('config.ini')