add find_packages in setup.py

Author: noahgolmant

setup.py

@@ -1,10 +1,14 @@
 """setup.py for hessian_eigenthings"""
 
-from setuptools import setup
+from setuptools import setup, find_packages
 
 install_requires = [
  'numpy>=0.14',
  'torch>=0.4'
 ]
 
-setup(name="hessian_eigenthings", author="Noah Golmant", install_requires=install_requires)
+setup(name="hessian_eigenthings",
+ author="Noah Golmant",
+ install_requires=install_requires,
+ packages=find_packages(),
+ description='Eigendecomposition of model Hessians in PyTorch!')