From 81fda3ec07bcfc65809404a53a82c15a444823bf Mon Sep 17 00:00:00 2001 From: Anmol-Srivastava Date: Fri, 10 Nov 2023 14:11:49 -0500 Subject: [PATCH] move dependencies to experimental option, document in notebook --- docs/source/examples/nlp/wilds_datasets.ipynb | 1 + setup.cfg | 4 +++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/docs/source/examples/nlp/wilds_datasets.ipynb b/docs/source/examples/nlp/wilds_datasets.ipynb index b26cbfc..9787ac8 100644 --- a/docs/source/examples/nlp/wilds_datasets.ipynb +++ b/docs/source/examples/nlp/wilds_datasets.ipynb @@ -16,6 +16,7 @@ "- transform functions are very slow, on even moderate batch sizes\n", "- detector design is not generalized, and may not work on streaming problems, or with data representations of different types/shapes\n", "- some warnings below are not addressed\n", + "- if not present, `toolz`, `tensorflow`, and `transformers` must be added via the `experimental` install option, and are not included by default\n", "\n", "## Imports\n", "\n", diff --git a/setup.cfg b/setup.cfg index 4c14f9a..48d741d 100644 --- a/setup.cfg +++ b/setup.cfg @@ -28,11 +28,13 @@ install_requires = pandas scipy scikit-learn + +[options.extras_require] +experimental = toolz tensorflow transformers -[options.extras_require] wilds = wilds matplotlib