diff --git a/src/typings/dataset.py b/src/typings/dataset.py
index 0a52f8e..59f7e5d 100644
--- a/src/typings/dataset.py
+++ b/src/typings/dataset.py
@@ -45,7 +45,9 @@ def load_data(self) -> Tuple[NumpyDataset, NumpyDataset]:
 
         return (x_train, y_train), (x_test, y_test)
 
-    def dataset(self, shuffle: int = 10000, batch: int = 32) -> Tuple[TestSet, TrainSet]:
+    def dataset(
+        self, shuffle: int = 10000, batch: int = 32
+    ) -> Tuple[TestSet, TrainSet]:
         (x_train, y_train), (x_test, y_test) = self.load_data()
 
         train_ds = (