images

Author: jph00

MANIFEST.in

@@ -3,3 +3,5 @@ include LICENSE
 include CONTRIBUTING.md
 include README.md
 recursive-exclude * __pycache__
+include fastbook/images/*.jpg
+

fastbook/__init__.py

@@ -1,8 +1,9 @@
 __version__ = "0.0.9"
+import matplotlib as mpl, pkgutil
 from fastai.vision.all import *
 from pandas.api.types import CategoricalDtype
 from scipy.cluster import hierarchy as hc
-import matplotlib as mpl
+from io import StringIO, BytesIO
 
 try: from ipywidgets import widgets
 except ModuleNotFoundError: warn("Missing `ipywidgets` - please install it")
@@ -71,3 +72,6 @@ def cluster_columns(df, figsize=(10,6), font_size=12):
     hc.dendrogram(z, labels=df.columns, orientation='left', leaf_font_size=font_size)
     plt.show()
 
+def image_cat (): return BytesIO(pkgutil.get_data('fastbook', 'images/cat.jpg'))
+def image_bear(): return BytesIO(pkgutil.get_data('fastbook', 'images/grizzly.jpg'))
+

setup.py

@@ -41,9 +41,7 @@
     packages = setuptools.find_packages(),
     include_package_data = True,
     install_requires = requirements,
-    extras_require = {
-        'dev': dev_requirements
-    },
+    extras_require = { 'dev': dev_requirements },
     python_requires  = '>=' + cfg['min_python'],
     long_description = open('README.md').read(),
     long_description_content_type = 'text/markdown',