Fix ModuleNotFoundError of line_profiler (#13)

In 0.2.0, line_profiler.py becomes sub directory, but not listed in `packages`.
Stonesjtu · Jun 18, 2020 · 1e87741 · 1e87741
1 parent ff71e73
commit 1e87741
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,4 +1,4 @@
-from setuptools import setup
+from setuptools import setup, find_packages
 
 try:
     long_description = open('README.md').read()
@@ -33,5 +33,5 @@
         'ipython': ['IPython>=0.13'],
         'test': ['pytest'],
     },
-    packages=['pytorch_memlab'],
+    packages=find_packages(),
 )