From 8c24c9ec2757c66854ef5fbaed89ef712c2992fe Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 26 Feb 2025 00:02:54 +0100
Subject: [PATCH] Fix CI

---
 face_swapper/src/training.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py
index dfbf9c6..a8cc6f6 100644
--- a/face_swapper/src/training.py
+++ b/face_swapper/src/training.py
@@ -9,7 +9,6 @@ from lightning import Trainer
 from lightning.pytorch.callbacks import ModelCheckpoint
 from lightning.pytorch.loggers import TensorBoardLogger
 from torch import Tensor, nn
-from torch.optim import Optimizer
 from torch.utils.data import DataLoader, Dataset, random_split
 
 from .dataset import DynamicDataset