From 6fa8d6b6eb5c770ecfd5f1c666fd67bdfb540ecd Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 5 Mar 2025 12:57:15 +0100 Subject: [PATCH] Add basic test for aad and unet --- .github/workflows/ci.yml | 2 +- face_swapper/tests/test_networks.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index d22407d..b6ced3e 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: pytest face_swapper/tests diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py index e1d4381..18ce42d 100644 --- a/face_swapper/tests/test_networks.py +++ b/face_swapper/tests/test_networks.py @@ -1,8 +1,8 @@ import torch import pytest -from face_swapper.src.networks.aad import AAD -from face_swapper.src.networks.unet import UNet +from ..src.networks.aad import AAD +from ..src.networks.unet import UNet @pytest.mark.parametrize('output_size', [ 256 ])