Crazy fix

This commit is contained in:
henryruhs committed 2025-03-11 14:43:10 +01:00
1 parent 4a2559c866
commit 0e8207ccc8
1 file changed
+1
+1
View File
@@ -29,4 +29,5 @@ 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)