Remove pytorch_lightning requirement from setup.py

This commit is contained in:
blackfly 2020-03-30 13:29:34 +02:00
parent 5aedc68527
commit cab2d5e77c

View File

@ -24,7 +24,6 @@ setup(name='prototorch',
install_requires=[
'torch>=1.3.1',
'torchvision>=0.4.2',
'pytorch-lightning>=0.6.0',
'numpy>=1.9.1',
'matplotlib',
'sklearn',
@ -45,6 +44,7 @@ setup(name='prototorch',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Operating System :: OS Independent',
'Topic :: Software Development :: Libraries',
'Topic :: Software Development :: Libraries :: Python Modules'