Install script changes.

xiaoxiawu-microsoft · Mar 19, 2021 · bbe1169 · bbe1169
1 parent c51c295
commit bbe1169
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -1 +0,0 @@
-include README.rst

diff --git a/setup.py b/setup.py
@@ -11,8 +11,8 @@ def readme():
 setup(
     name="nn_pruning",
     version="0.1",
-    description="nn_pruning is a Python package for pruning Transformers",
-    long_description=readme(),
+    description="nn_pruning is a python package for pruning PyTorch models.",
+    long_description="nn_pruning is a python package for pruning PyTorch models.",
     classifiers=[
         "Development Status :: 3 - Alpha",
         "License :: OSI Approved :: MIT License",
@@ -24,7 +24,7 @@ def readme():
     author="",
     author_email="",
     license="MIT",
-    packages=["nn_pruning"],
+    packages=['nn_pruning', 'nn_pruning.modules'],
     install_requires=["click", "transformers>=4.3.0", "torch>=1.6", "scikit-learn>=0.24"],
     extras_require=extras,
     test_suite="nose.collector",