Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixing setup

This commit is contained in:
William Falcon 2019-04-21 13:09:06 -04:00
parent ee51d7b7bc
commit e9bcbc2318
2 changed files with 2 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch-lightning/trainer_main.py
View File

@ -9,7 +9,6 @@ from pytorch_lightning.utils.arg_parse import add_default_args
from time import sleep from time import sleep
from pytorch_lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint from pytorch_lightning.utils.pt_callbacks import EarlyStopping, ModelCheckpoint
SEED = 2334 SEED = 2334
torch.manual_seed(SEED) torch.manual_seed(SEED)
np.random.seed(SEED) np.random.seed(SEED)

30
setup.py
View File

@ -1,13 +1,13 @@
#!/usr/bin/env python #!/usr/bin/env python
from setuptools import setup, find_packages, os from setuptools import setup, find_packages
# https://packaging.python.org/guides/single-sourcing-package-version/ # https://packaging.python.org/guides/single-sourcing-package-version/
# http://blog.ionelmc.ro/2014/05/25/python-packaging/ # http://blog.ionelmc.ro/2014/05/25/python-packaging/
setup( setup(
name="pytorch-lightning", name="pytorch-lightning",
version='0.1.1', version='0.1.2',
description="The Keras for ML researchers using PyTorch", description="The Keras for ML researchers using PyTorch",
author="William Falcon", author="William Falcon",
author_email="waf2107@columbia.edu", author_email="waf2107@columbia.edu",
@ -21,33 +21,7 @@ setup(
"tqdm", "tqdm",
"test-tube", "test-tube",
], ],
extras_require={
"dev": [
"black ; python_version>='3.6'",
"coverage",
"isort",
"pytest",
"pytest-cov<2.6.0",
"pycodestyle",
"sphinx",
"nbsphinx",
"ipython>=5.0",
"jupyter-client",
]
},
packages=find_packages(), packages=find_packages(),
classifiers=[
"Development Status :: 4 - Beta",
"Intended Audience :: Education",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Programming Language :: Python",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.5",
"Programming Language :: Python :: 3.6",
"Programming Language :: Python :: 3.7",
],
long_description=open("README.md", encoding="utf-8").read(), long_description=open("README.md", encoding="utf-8").read(),
include_package_data=True, include_package_data=True,
zip_safe=False, zip_safe=False,