From 10ec3040a2d2c018ebb6f6facaef4fbb580e44f4 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Mon, 30 Mar 2026 11:09:14 +0530 Subject: [PATCH] [ci] move to assert instead of self.Assert* (#13366) move to assert instead of self.Assert* --- tests/models/autoencoders/testing_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/autoencoders/testing_utils.py b/tests/models/autoencoders/testing_utils.py index 8ae362ac2e..68b65dc354 100644 --- a/tests/models/autoencoders/testing_utils.py +++ b/tests/models/autoencoders/testing_utils.py @@ -44,9 +44,9 @@ class AutoencoderTesterMixin: if isinstance(output, dict): output = output.to_tuple()[0] - self.assertIsNotNone(output) + assert output is not None expected_shape = inputs_dict["sample"].shape - 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" def test_enable_disable_tiling(self): if not hasattr(self.model_class, "enable_tiling"):