Use find_packages() for setup.py

1 files changed, 2 insertions, 9 deletions

diff --git a/setup.py b/setup.py
index 5fc2417..8a2a08a 100644
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,6 @@
 import sys
 import os
-from setuptools import setup, Command
+from setuptools import setup, Command, find_packages
 
 # Pull version from source without importing
 # since we can't import something we haven't built yet :)
@@ -37,14 +37,7 @@ setup(
 
     tests_require=test_require,
     cmdclass={"test": Tox},
-
-    packages=[
-        "kafka",
-        "kafka.consumer",
-        "kafka.partitioner",
-        "kafka.producer",
-    ],
-
+    packages=find_packages(exclude=['test']),
     author="Dana Powers",
     author_email="dana.powers@gmail.com",
     url="https://github.com/dpkp/kafka-python",