docs tweak

dennisprangle · May 29, 2021 · 2b89b16 · 2b89b16
1 parent b524042
commit 2b89b16
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -31,7 +31,10 @@ def read_package_variable(key):
     author_email='[email protected]',
     packages=find_packages(),
     install_requires=['torch>=1.5.1', 'numpy', 'scikit-learn', 'matplotlib', 'termcolor==1.1.0', 'pyzmq>=19.0.0', 'flatbuffers==1.12', 'pydotplus==2.0.2', 'pyyaml>=3.13'],
-    extras_require={'dev': ['pytest', 'pytest-cov', 'pytest-xdist', 'docker', 'sphinx', 'sphinx_rtd_theme', 'nbsphinx', 'sphinxcontrib-bibtex']},
+    extras_require={
+        'dev': ['pytest', 'pytest-cov', 'pytest-xdist', 'docker'],
+        'docs': ['sphinx==3.2.1', 'sphinx_rtd_theme==0.5.2', 'jupyter-sphinx==0.3.2', 'myst-nb==0.12.3']
+        },
     url='https://github.com/pyprob/pyprob',
     classifiers=['License :: OSI Approved :: BSD License', 'Programming Language :: Python :: 3'],
     license='BSD',