diff --git a/.readthedocs.yml b/.readthedocs.yml new file mode 100644 index 000000000..228fc8af8 --- /dev/null +++ b/.readthedocs.yml @@ -0,0 +1,8 @@ +formats: + - none +requirements_file: requirements.txt +python: + pip_install: true + extra_requirements: + - tests + - docs diff --git a/requirements.txt b/requirements.txt index 0b6a60849..473918987 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ scipy>=0.13.3 numpy>=1.8.2 -scikit-learn>=0.20 +scikit-learn>=0.20rc1 diff --git a/setup.py b/setup.py index af38fb666..6fe2c2d49 100755 --- a/setup.py +++ b/setup.py @@ -36,6 +36,21 @@ 'Programming Language :: Python :: 2.7', 'Programming Language :: Python :: 3.6', 'Programming Language :: Python :: 3.7'] +EXTRAS_REQUIRE = { + 'tests': [ + 'pytest', + 'pytest-cov'], + 'docs': [ + 'sphinx', + 'sphinx-gallery', + 'sphinx_rtd_theme', + 'numpydoc', + 'matplotlib', + 'pandas', + 'keras', + 'tensorflow' + ] +} setup(name=DISTNAME, @@ -50,4 +65,5 @@ zip_safe=False, # the package can run out of an .egg file classifiers=CLASSIFIERS, packages=find_packages(), - install_requires=INSTALL_REQUIRES) + install_requires=INSTALL_REQUIRES, + extras_require=EXTRAS_REQUIRE)