Ready to release the new version

YeonwooSung · Apr 24, 2021 · e1b4137 · e1b4137
1 parent fec7022
commit e1b4137
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -9,9 +9,9 @@
 setup(
     name='pyglom',
     packages=find_packages(),
-    version='0.0.1',
+    version='0.0.2',
     license='MIT',
-    description='Pytorch implementation of GLOM',
+    description='PyTorch implementation of GLOM. PyTorch-Lightning version is also available.',
     author='Yeonwoo Sung',
     author_email='[email protected]',
     url='https://github.com/YeonwooSung/GLOM',
@@ -28,6 +28,7 @@
     ],
     install_requires=[
         'einops>=0.3',
+        "pytorch-lightning==0.8.5",
         'torch>=1.6'
     ],
     classifiers=[