From 798ff877369a699637b803e4029e6c0bc23636b9 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sun, 16 Mar 2025 15:18:28 +0100 Subject: [PATCH] Adjust naming --- face_swapper/tests/test_networks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py index c0551be..654a172 100644 --- a/face_swapper/tests/test_networks.py +++ b/face_swapper/tests/test_networks.py @@ -52,6 +52,6 @@ def test_mask_net(output_size : int) -> None: target_tensor = torch.randn(1, 3, output_size, output_size) target_feature = torch.randn(1, 64, output_size, output_size) - output_tensor = masker(target_tensor, target_feature) + output_mask = masker(target_tensor, target_feature) - assert output_tensor.shape == (1, 1, output_size, output_size) + assert output_mask.shape == (1, 1, output_size, output_size)