diff --git a/flash/core/data/data_module.py b/flash/core/data/data_module.py
index bf8be1ba42..5e588ba698 100644
--- a/flash/core/data/data_module.py
+++ b/flash/core/data/data_module.py
@@ -595,4 +595,4 @@ def _split_train_val(
         return (
             SplitDataset(train_dataset, train_indices, use_duplicated_indices=True),
             SplitDataset(train_dataset, val_indices, use_duplicated_indices=True),
-        )
\ No newline at end of file
+        )
diff --git a/tests/core/data/test_data_module.py b/tests/core/data/test_data_module.py
index 20594dbda7..f0e9f47a85 100644
--- a/tests/core/data/test_data_module.py
+++ b/tests/core/data/test_data_module.py
@@ -469,4 +469,4 @@ def test_val_split():
     )
 
     assert len(datamodule.train_dataset) == 80
-    assert len(datamodule.val_dataset) == 20
\ No newline at end of file
+    assert len(datamodule.val_dataset) == 20