(#4) Refactor: style

betarixm · Mar 24, 2022 · c5d237b · c5d237b
1 parent 5e74e22
commit c5d237b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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 = (