diff --git a/datasets/e2e/tensorflow/tensorflow_test.py b/datasets/e2e/tensorflow/tensorflow_test.py
index e041bcb8f8cc..5e21b5d6386f 100644
--- a/datasets/e2e/tensorflow/tensorflow_test.py
+++ b/datasets/e2e/tensorflow/tensorflow_test.py
@@ -38,8 +38,7 @@ class FdsToTensorFlow(unittest.TestCase):
 
     @classmethod
     def setUpClass(cls):
-        """Disable progress bar to keep the log clean.
-        """
+        """Disable progress bar to keep the log clean."""
         disable_progress_bar()
 
     def _create_tensorflow_dataset(self, batch_size: int) -> tf.data.Dataset: