Crazy fix

This commit is contained in:
henryruhs
2025-03-05 14:36:07 +01:00
parent 4a2559c866
commit 0e8207ccc8
+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)