From c5d237be3724678c89f95dae117bd8189d54fc5e Mon Sep 17 00:00:00 2001 From: beta Date: Thu, 24 Mar 2022 12:39:24 +0900 Subject: [PATCH] (#4) Refactor: style --- src/typings/dataset.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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 = (