Remove unused return

lightly-ai · Jan 26, 2024 · 3980788 · 3980788
1 parent 81f44dc
commit 3980788
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/modules/test_masked_autoencoder.py b/tests/models/modules/test_masked_autoencoder.py
@@ -93,7 +93,7 @@ def test_forward_cuda(self):
 )
 class TestMAEDecoder(unittest.TestCase):
     def test_init(self):
-        return MAEDecoder(
+        MAEDecoder(
             seq_length=50,
             num_layers=2,
             num_heads=4,