diff --git a/test/test_linalg.py b/test/test_linalg.py index 3fb6c8696c00a..f5aed1202a1b3 100644 --- a/test/test_linalg.py +++ b/test/test_linalg.py @@ -109,12 +109,12 @@ def get_tunableop_untuned_filename(): class TestLinalg(TestCase): def setUp(self): - super(self.__class__, self).setUp() + super().setUp() torch.backends.cuda.matmul.allow_tf32 = False def tearDown(self): torch.backends.cuda.matmul.allow_tf32 = True - super(self.__class__, self).tearDown() + super().tearDown() @contextlib.contextmanager def _tunableop_ctx(self):