[trainer / deepspeed] fix hyperparameter_search (#16740)

* [trainer / deepspeed] fix hyperparameter_search * require optuna * style * oops * add dep in the right place * create deepspeed-testing dep group * Trigger CI
2022-04-14 17:24:38 -07:00
parent 1b7de41a07
commit ce2fef2ad2
6 changed files with 36 additions and 5 deletions
--- a/setup.py
+++ b/setup.py
@@ -290,6 +290,8 @@ extras["testing"] = (
    + extras["modelcreation"]
 )

+extras["deepspeed-testing"] = extras["deepspeed"] + extras["testing"] + extras["optuna"]
+
 extras["quality"] = deps_list("black", "isort", "flake8", "GitPython", "hf-doc-builder")

 extras["all"] = (