Skip to content

Commit

Permalink
Address pep-8 violations in dataset init
Browse files Browse the repository at this point in the history
  • Loading branch information
JMGaljaard committed Sep 6, 2022
1 parent 1d28dd2 commit d4a630e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fltk/datasets/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
from fltk.datasets.mnist import MNIST
from fltk.util.config.definitions import Dataset


def get_train_loader_path(name: Dataset) -> str:
paths = {
Dataset.cifar10: 'data_loaders/cifar10/train_data_loader.pickle',
Expand All @@ -22,4 +23,3 @@ def get_test_loader_path(name: Dataset) -> str:
Dataset.mnist: 'data_loaders/mnist/test_data_loader.pickle',
}
return paths[name]

0 comments on commit d4a630e

Please sign in to comment.