From 0aee9edce687bdd1fdab93793df7c170a2120b29 Mon Sep 17 00:00:00 2001 From: Connor Olding Date: Mon, 2 Jul 2018 15:50:05 +0200 Subject: [PATCH] fix downloading and verification of first files this code would've worked if the first element of `names` were the prefix i.e. if it acted on `meta` instead of `names`. pretty embarassing! --- mnists/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mnists/__init__.py b/mnists/__init__.py index bb49487..59fb8e4 100644 --- a/mnists/__init__.py +++ b/mnists/__init__.py @@ -160,7 +160,7 @@ def prepare(dataset="mnist", return_floats=True, return_onehot=True, prefix, names = meta[0], meta[1:] names = [os.path.join(prefix, name) for name in names] train_images, train_labels, test_images, test_labels = names - images_and_labels = names[1:] + images_and_labels = names logger.debug("Filenames chosen for %s: %s, %s, %s, %s", dataset, train_images, train_labels, test_images, test_labels)