def _prepare_extras():
extras = {
'loggers': setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name='loggers.txt'),
'models': setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name='models.txt'),
'test': setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name='test.txt'),
}
extras['extra'] = extras['models'] + extras['loggers']
extras['dev'] = extras['extra'] + extras['test']
return extras
def _prepare_extras():
extras = {
"loggers": setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name="loggers.txt"),
"models": setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name="models.txt"),
"test": setup_tools._load_requirements(path_dir=_PATH_REQUIRE, file_name="test.txt"),
}
extras["extra"] = extras["models"] + extras["loggers"]
extras["dev"] = extras["extra"] + extras["test"]
return extras
def _prepare_extras():
extras = {
'loggers':
_load_requirements(path_dir=os.path.join(_PATH_ROOT, 'requirements'),
file_name='loggers.txt'),
'models':
_load_requirements(path_dir=os.path.join(_PATH_ROOT, 'requirements'),
file_name='models.txt'),
'test':
_load_requirements(path_dir=os.path.join(_PATH_ROOT, 'requirements'),
file_name='test.txt'),
}
extras['extra'] = extras['models'] + extras['loggers']
extras['dev'] = extras['extra'] + extras['test']
return extras
version=pl_bolts.__version__,
description=pl_bolts.__docs__,
author=pl_bolts.__author__,
author_email=pl_bolts.__author_email__,
url=pl_bolts.__homepage__,
download_url='https://github.com/PyTorchLightning/lightning-bolts',
license=pl_bolts.__license__,
packages=find_packages(exclude=['tests', 'docs']),
long_description=_load_readme_description(_PATH_ROOT),
long_description_content_type='text/markdown',
include_package_data=True,
zip_safe=False,
keywords=['deep learning', 'pytorch', 'AI'],
python_requires='>=3.6',
setup_requires=['wheel'],
install_requires=_load_requirements(_PATH_ROOT),
extras_require=_prepare_extras(),
project_urls={
"Bug Tracker":
"https://github.com/PyTorchLightning/lightning-bolts/issues",
"Documentation": "https://lightning-bolts.rtfd.io/en/latest/",
"Source Code": "https://github.com/PyTorchLightning/lightning-bolts",
},
classifiers=[
'Environment :: Console',
'Natural Language :: English',
# How mature is this project? Common values are
# 3 - Alpha, 4 - Beta, 5 - Production/Stable
'Development Status :: 4 - Beta',
# Indicate who your project is intended for
'Intended Audience :: Developers',
version=info.__version__,
description=info.__docs__,
author=info.__author__,
author_email=info.__author_email__,
url=info.__homepage__,
download_url='https://github.com/PyTorchLightning/lightning-bolts',
license=info.__license__,
packages=find_packages(exclude=['tests', 'docs']),
long_description=long_description,
long_description_content_type='text/markdown',
include_package_data=True,
zip_safe=False,
keywords=['deep learning', 'pytorch', 'AI'],
python_requires='>=3.6',
setup_requires=['wheel'],
install_requires=setup_tools._load_requirements(_PATH_ROOT),
extras_require=_prepare_extras(),
project_urls={
"Bug Tracker":
"https://github.com/PyTorchLightning/lightning-bolts/issues",
"Documentation": "https://lightning-bolts.rtfd.io/en/latest/",
"Source Code": "https://github.com/PyTorchLightning/lightning-bolts",
},
classifiers=[
'Environment :: Console',
'Natural Language :: English',
# How mature is this project? Common values are
# 3 - Alpha, 4 - Beta, 5 - Production/Stable
'Development Status :: 4 - Beta',
# Indicate who your project is intended for
'Intended Audience :: Developers',