From 3980788b235d1010129ba1ce3149f7c1dea82d6b Mon Sep 17 00:00:00 2001
From: guarin <guarin@lightly.ai>
Date: Fri, 26 Jan 2024 09:49:59 +0000
Subject: [PATCH] Remove unused return

---
 tests/models/modules/test_masked_autoencoder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/models/modules/test_masked_autoencoder.py b/tests/models/modules/test_masked_autoencoder.py
index f96985a7b..42210bfdb 100644
--- 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,