Setup fix (#186)

* setup_fix * fix setup
rll · Nov 20, 2017 · b3a2899 · b3a2899
1 parent 9adb7e6
commit b3a2899
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,8 +1,9 @@
 # setup.py
-from setuptools import setup
+from setuptools import setup,find_packages
 
 setup(
     name='rllab',
+    packages=[package for package in find_packages()
+                if package.startswith('rllab')],
     version='0.1.0',
-    packages=['rllab'],
 )