Add basic test for aad and unet

This commit is contained in:
henryruhs
2025-03-05 12:57:15 +01:00
parent 72371b9f11
commit 6fa8d6b6eb
2 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -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
+2 -2
View File
@@ -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 ])