From 7264884ff94041887c00033b94cb2f2114c0cf46 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Thu, 13 Feb 2025 19:13:49 +0100
Subject: [PATCH] Fix CI

---
 face_swapper/src/networks/unet.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/face_swapper/src/networks/unet.py b/face_swapper/src/networks/unet.py
index 6454455..ebf88ba 100644
--- a/face_swapper/src/networks/unet.py
+++ b/face_swapper/src/networks/unet.py
@@ -11,7 +11,7 @@ class UNet(nn.Module):
 		self.up = self.create_up()
 
 	@staticmethod
-	def create_down():
+	def create_down() -> nn.ModuleList:
 		return nn.ModuleList(
 		[
 			Down(3, 32),
@@ -24,7 +24,7 @@ class UNet(nn.Module):
 		])
 
 	@staticmethod
-	def create_up():
+	def create_up() -> nn.ModuleList:
 		return nn.ModuleList(
 		[
 			Up(1024, 1024),