diff --git a/setup.py b/setup.py index 0e55cc0..15a8d08 100644 --- a/setup.py +++ b/setup.py @@ -110,9 +110,9 @@ def default_setup_args(*, version): extras_require = dict() full_requirements = ['matplotlib', 'autogluon.tabular', 'torch', 'xgboost', 'jupyterlab'] -notebook_requirements = full_requirements + ['fairlearn'] -test_requirements = notebook_requirements + ["tox", "pytest", "pytest-cov", 'flake8', - 'linkcheckmd', 'ipynbcompress',] +notebook_requirements = full_requirements + ['fairlearn', 'fairret'] +test_requirements = notebook_requirements + ["tox", "pytest", "pytest-cov", 'flake8', 'tabulate', + 'linkcheckmd', 'ipynbcompress', 'nbmake'] full_requirements = list(set(full_requirements)) notebook_requirements = list(set(notebook_requirements))