Skip to content

Commit 0840b27

Browse files
Format MAISI ControlNet test
Signed-off-by: ugbotueferhire <ugbotueferhire@gmail.com>
1 parent c60d5f2 commit 0840b27

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

tests/apps/maisi/networks/test_controlnet_maisi.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -196,11 +196,7 @@ def test_modality_input_missing(self):
196196
)
197197
with self.assertRaisesRegex(ValueError, "modality_tensor should be provided"):
198198
with eval_mode(net):
199-
net.forward(
200-
torch.rand((1, 1, 16, 16)),
201-
torch.randint(0, 1000, (1,)).long(),
202-
torch.rand((1, 1, 32, 32)),
203-
)
199+
net.forward(torch.rand((1, 1, 16, 16)), torch.randint(0, 1000, (1,)).long(), torch.rand((1, 1, 32, 32)))
204200

205201

206202
if __name__ == "__main__":

0 commit comments

Comments
 (0)