Skip to content

Commit 54dd32a

Browse files
committed
fix a bug in testing only returning loss_style
1 parent c1ace45 commit 54dd32a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_losses/test_gen_auxiliary_loss.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def test_only_style_loss(self):
109109
data_info=self.data_info, perceptual_weight=0)
110110
loss_style = perceptual_loss(dict(fake_imgs=pred, real_imgs=target))
111111
assert loss_style.shape == ()
112-
assert perceptual_loss.perceptual_weight is None
112+
assert perceptual_loss.perceptual_weight == 0
113113

114114
def test_with_different_layer_weights(self):
115115
unknown_h, unknown_w = (32, 32)

0 commit comments

Comments
 (0)