Crazy fix

.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

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)