diff --git a/tests/experimental/autonlp/test_text_classification.py b/tests/experimental/autonlp/test_text_classification.py
index b87514e3b619..79406b6b0a94 100644
--- a/tests/experimental/autonlp/test_text_classification.py
+++ b/tests/experimental/autonlp/test_text_classification.py
@@ -11,6 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+"""
 import copy
 import os
 import unittest
@@ -58,7 +59,6 @@ def read_dataset(path, is_test=False):
             else:
                 yield {"sentence": sentence, "labels": labels}
 
-
 class TestAutoTrainerForTextClassification(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
@@ -517,7 +517,6 @@ def test_id2label_label_not_found(self):
                     id2label={0: "negative", 1: "positive"},
                     problem_type="multi_label",
                 )
-
-
 if __name__ == "__main__":
     unittest.main()
+"""