Skip to content

Commit 10ec304

Browse files
authored
[ci] move to assert instead of self.Assert* (#13366)
move to assert instead of self.Assert*
1 parent f2be8bd commit 10ec304

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/models/autoencoders/testing_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,9 @@ def test_forward_with_norm_groups(self):
4444
if isinstance(output, dict):
4545
output = output.to_tuple()[0]
4646

47-
self.assertIsNotNone(output)
47+
assert output is not None
4848
expected_shape = inputs_dict["sample"].shape
49-
self.assertEqual(output.shape, expected_shape, "Input and output shapes do not match")
49+
assert output.shape == expected_shape, "Input and output shapes do not match"
5050

5151
def test_enable_disable_tiling(self):
5252
if not hasattr(self.model_class, "enable_tiling"):

0 commit comments

Comments
 (0)