Revise setup.py

suhigh · Sep 27, 2017 · aca53bd · aca53bd
1 parent cbc1761
commit aca53bd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -13,6 +13,9 @@
     # TODO: Different OS have different requirements
 ]
 
+required_packages = find_packages()
+required_packages.append('labelImg')
+
 setup(
     name='labelImg',
     version='1.5.1',
@@ -21,7 +24,8 @@
     author="TzuTa Lin",
     author_email='[email protected]',
     url='https://github.com/tzutalin/labelImg',
-    packages=find_packages(),
+    package_dir={'labelImg': '.'},
+    packages=required_packages,
     entry_points={
         'console_scripts': [
             'labelImg=labelImg.labelImg:main'
@@ -46,4 +50,5 @@
         'Programming Language :: Python :: 3.5',
         'Programming Language :: Python :: 3.6',
     ],
+    package_data={'data/predefined_classes.txt': ['data/predefined_classes.txt']}
 )