From d5c51a90e890e8473b34231a6e403188185c41d1 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 5 Mar 2025 13:10:14 +0100
Subject: [PATCH] Add basic test for aad and unet

---
 face_swapper/tests/test_networks.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py
index 18ce42d..e1d4381 100644
--- a/face_swapper/tests/test_networks.py
+++ b/face_swapper/tests/test_networks.py
@@ -1,8 +1,8 @@
 import torch
 import pytest
 
-from ..src.networks.aad import AAD
-from ..src.networks.unet import UNet
+from face_swapper.src.networks.aad import AAD
+from face_swapper.src.networks.unet import UNet
 
 
 @pytest.mark.parametrize('output_size', [ 256 ])