remove of the matplotlib import

setup.py

@@ -16,9 +16,9 @@ setup(
     name='scikit-luc',
     # You can specify all the packages manually or use the find_package
     # function
-    # packages=find_packages(exclude=['doc', 'examples']),
+    packages=find_packages(exclude=['doc', 'examples']),
     # See PEP440 for defining a proper version number
-    version='1.02',
+    version='1.6',
     # Small description of the package
     description='Science-Kit developed by Luc Giffon for Luc Giffon during his PhD thesis.',
     # Long description

skluc/__init__.py

@@ -2,4 +2,4 @@
 This package contains some usefull functions for the PhD of Luc Giffon.
 """
 
-__version__ = "1.02"
+__version__ = "1.6"
\ No newline at end of file

skluc/mldatasets.py

@@ -13,7 +13,6 @@ import logging
 import tarfile
 import pickle
 # --- installed packages
-import matplotlib.pyplot as plt
 import daiquiri
 import numpy as np
 
@@ -97,8 +96,6 @@ class Dataset:
 
         :return: None
         """
-        # the download method shouldn't be called if the target directory tree doesn't exist.
-        assert os.path.exists(self.s_download_dir)
         self.create_directory_tree()
         if not check_files(self.l_filepaths):
             logger.info("Files need to be downloaded")
@@ -166,7 +163,7 @@ class MnistDataset(Dataset):
         with gzip.open(fname_img, 'rb') as fimg:
             magic, num, rows, cols = struct.unpack(">IIII", fimg.read(16))
             img = np.fromstring(fimg.read(), dtype=np.uint8)
-            img = img.reshape(len(lbl), rows, cols)
+            img = img.reshape(len(lbl), -1)
 
         return img, lbl