Fix an error with subpackages

Rishit-dagli · Aug 12, 2021 · 17a378b · 17a378b
1 parent e31d26f
commit 17a378b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 import os.path
 
-from setuptools import setup
+from setuptools import setup, find_packages
 
 
 def read(rel_path: str) -> str:
@@ -27,7 +27,7 @@ def get_version(rel_path: str) -> str:
     name="tf-watcher",
     version=get_version("tfwatcher/version.py"),
     description="Monitor your TensorFlow model training on mobile devices, especially for Google Colab",
-    packages=["tfwatcher"],
+    packages=find_packages(),
     long_description=long_description,
     long_description_content_type="text/markdown",
     classifiers=[