adds subpackages to setup.py

tfjgeorge · Apr 14, 2020 · 366d5f5 · 366d5f5
1 parent c386eea
commit 366d5f5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -7,6 +7,8 @@
       author='tfjgeorge',
       author_email='tfjgeorge@gmail.com',
       license='MIT',
-      packages=['nngeometry'],
+      packages=['nngeometry',
+                'nngeometry.generator',
+                'nngeometry.object'],
       install_requires=['torch>=1.0.0'],
       zip_safe=False)