From e5f983b2bf1bb9a918d7cba845ce2ac7af457bf2 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 6 Mar 2025 22:28:54 +0100 Subject: [PATCH] Minor adjustment for test --- face_swapper/tests/test_networks.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py index 16305e8..96f1c6a 100644 --- a/face_swapper/tests/test_networks.py +++ b/face_swapper/tests/test_networks.py @@ -10,13 +10,16 @@ from face_swapper.src.networks.unet import UNet @pytest.mark.parametrize('output_size', [ 128, 256, 512 ]) def test_aad_with_unet(output_size : int) -> None: config_parser = ConfigParser() - config_parser['training.model.generator'] =\ + config_parser.read_dict( { - 'identity_channels': '512', - 'output_channels': str(output_size * 16), - 'output_size': str(output_size), - 'num_blocks': '2' - } + 'training.model.generator': + { + 'identity_channels': '512', + 'output_channels': str(output_size * 16), + 'output_size': str(output_size), + 'num_blocks': '2' + } + }) generator = AAD(config_parser).eval() encoder = UNet(config_parser).eval()