Added the submodules

wncc · Oct 7, 2023 · 07fbdd1 · 07fbdd1
1 parent 6b646ff
commit 07fbdd1
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,16 +1,21 @@
 from setuptools import setup
+from setuptools import find_packages
 
 setup(
     name='UniTrain',
-    version='0.2',
+    version='0.2.1',
     author='Ishan Upadhyay',
     author_email='[email protected]',
     description='A generalized training framework for Deep Learning Tasks',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',
     url='https://github.com/ishan121028/UniTrain',
     download_url='https://github.com/ishan121028/UniTrain/archive/v0.1.tar.gz',
-    packages=['UniTrain'],
+    packages=find_packages(
+        where='UniTrain',
+        include=['*'],    
+        ) ,
+    package_dir={'':'UniTrain'},
     install_requires=[
         'torch',
         'torchvision',