tighten requirements to specify tests and docs require

LouisDesdoigts · May 29, 2023 · 423976d · 423976d
1 parent 852200f
commit 423976d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 11 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,13 +1,4 @@
 # Core
 jax
 jaxlib
-zodiax==0.4.0
-
-# Extra
-optax
-matplotlib
-jupyter
-jupyterlab
-tqdm
-chainconsumer
-# nummpyro
+zodiax==0.4.0
diff --git a/setup.py b/setup.py
@@ -22,7 +22,9 @@ def find_version(*file_paths):
 with open('requirements.txt') as f:
     install_requires = f.read().splitlines()
 # 2. What dependencies required to run the unit tests? (i.e. `pytest --remote-data`)
-# tests_require = ['pytest', 'pytest-cov', 'pytest-remotedata']
+tests_require = ['pytest']
+docs_require = ['optax', 'matplotlib', 'jupyter', 'jupyterlab', 'tqdm', 
+    'chainconsumer', 'nummpyro']
 
 
 setuptools.setup(
@@ -41,6 +43,8 @@ def find_version(*file_paths):
     },
 
     install_requires=install_requires,
+    tests_require=tests_require,
+    extras_require={'docs': docs_require},
 
     classifiers=[
         "Programming Language :: Python :: 3",