Include ImageNet json in the package

MisaOgura · Jun 13, 2019 · 09ba0d7 · 09ba0d7
1 parent 7c85304
commit 09ba0d7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -0,0 +1 @@
+include flashtorch/utils/resources/imagenet_class_index.json
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
 with open('README.md', 'r') as fh:
     long_description = fh.read()
 
-version = '0.0.5'
+version = '0.0.6'
 
 setup(
     name='flashtorch',
@@ -16,12 +16,17 @@
     long_description_content_type='text/markdown',
     url='https://github.com/MisaOgura/flashtorch',
     packages=find_packages(exclude=['tests*']),
+    include_package_data=True,
+    package_data={'flashtorch.utils.resources':
+        ['imagenet_class_index.json']
+    },
     install_requires=[
         'matplotlib',
         'numpy',
         'Pillow',
         'torch',
         'torchvision',
+        'importlib_resources'
     ],
     classifiers=[
         'Programming Language :: Python :: 3',