We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f2be8bd commit 10ec304Copy full SHA for 10ec304
tests/models/autoencoders/testing_utils.py
@@ -44,9 +44,9 @@ def test_forward_with_norm_groups(self):
44
if isinstance(output, dict):
45
output = output.to_tuple()[0]
46
47
- self.assertIsNotNone(output)
+ assert output is not None
48
expected_shape = inputs_dict["sample"].shape
49
- self.assertEqual(output.shape, expected_shape, "Input and output shapes do not match")
+ assert output.shape == expected_shape, "Input and output shapes do not match"
50
51
def test_enable_disable_tiling(self):
52
if not hasattr(self.model_class, "enable_tiling"):
0 commit comments