From 3c5554c1c5b09ecc42d703fbd164c5bedb43e723 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 5 Mar 2025 14:32:27 +0100 Subject: [PATCH] Crazy fix --- face_swapper/tests/test_networks.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py index dbc1773..5f74f83 100644 --- a/face_swapper/tests/test_networks.py +++ b/face_swapper/tests/test_networks.py @@ -1,9 +1,13 @@ +import os import sys import pytest import torch -sys.path.append('..') +package_directory = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) +sys.path.append(package_directory) + +print(package_directory) from face_swapper.src.networks.aad import AAD from face_swapper.src.networks.unet import UNet