From f678aa8f7e4a4ee42140b50edc873a2fed357783 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 5 Mar 2025 14:41:15 +0100 Subject: [PATCH] Crazy fix --- .github/workflows/ci.yml | 2 +- face_swapper/tests/test_networks.py | 12 ++---------- 2 files changed, 3 insertions(+), 11 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index d22407d..e3ba555 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -31,4 +31,4 @@ jobs: python-version: '3.12' - run: pip install torch torchvision - run: pip install pytest - - run: pytest + - run: PYTHONPATH=/Users/runner/work/facefusion-labs/facefusion-labs pytest diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py index 82c192f..f848884 100644 --- a/face_swapper/tests/test_networks.py +++ b/face_swapper/tests/test_networks.py @@ -1,15 +1,8 @@ -import os -import sys - import pytest import torch -package_directory = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) -sys.path.append(package_directory) - -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 ]) def test_aad_with_unet(output_size : int) -> None: @@ -29,5 +22,4 @@ def test_aad_with_unet(output_size : int) -> None: target_attributes = encoder(target_tensor) output_tensor = generator(source_tensor, target_attributes) - assert package_directory == None assert output_tensor.shape == (1, 3, output_size, output_size)