fix setup.py: use find_packages to install all packages

rfeinman · Sep 27, 2022 · a252e30 · a252e30
1 parent 85b78ba
commit a252e30
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,4 +1,6 @@
-from setuptools import setup
+from setuptools import setup, find_packages
+
+packages = find_packages(exclude=("tests", "tests.*"))
 
 setup(
     name='pytorch-minimize',
@@ -8,7 +10,7 @@
     author='Reuben Feinman',
     author_email='reuben.feinman@nyu.edu',
     license='MIT Licence',
-    packages=['torchmin'],
+    packages=packages,
     zip_safe=False,
     install_requires=[
         'numpy>=1.18.0',