Skip to content

GitLab

Unverified Commit 8d171628 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Test optuna and ray (#11924)

parent af1a10bf
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 0 deletions
setup.py
View file @ 8d171628
...@@ -112,6 +112,7 @@ _deps = [ ...@@ -112,6 +112,7 @@ _deps = [
"onnxconverter-common", "onnxconverter-common",
"onnxruntime-tools>=1.4.2", "onnxruntime-tools>=1.4.2",
"onnxruntime>=1.4.0", "onnxruntime>=1.4.0",
"optuna",
"packaging", "packaging",
"parameterized", "parameterized",
"protobuf", "protobuf",
...@@ -121,6 +122,7 @@ _deps = [ ...@@ -121,6 +122,7 @@ _deps = [
"pytest-sugar", "pytest-sugar",
"pytest-xdist", "pytest-xdist",
"python>=3.6.0", "python>=3.6.0",
"ray",
"recommonmark", "recommonmark",
"regex!=2019.12.17", "regex!=2019.12.17",
"requests", "requests",
...@@ -239,6 +241,10 @@ extras["modelcreation"] = deps_list("cookiecutter") ...@@ -239,6 +241,10 @@ extras["modelcreation"] = deps_list("cookiecutter")
extras["sagemaker"] = deps_list("sagemaker") extras["sagemaker"] = deps_list("sagemaker")
extras["deepspeed"] = deps_list("deepspeed") extras["deepspeed"] = deps_list("deepspeed")
extras["fairscale"] = deps_list("fairscale") extras["fairscale"] = deps_list("fairscale")
extras["optuna"] = deps_list("optuna")
extras["ray"] = deps_list("ray")
extras["integrations"] = extras["optuna"] + extras["ray"]
extras["serving"] = deps_list("pydantic", "uvicorn", "fastapi", "starlette") extras["serving"] = deps_list("pydantic", "uvicorn", "fastapi", "starlette")
extras["speech"] = deps_list("soundfile", "torchaudio") extras["speech"] = deps_list("soundfile", "torchaudio")
...@@ -263,6 +269,7 @@ extras["all"] = ( ...@@ -263,6 +269,7 @@ extras["all"] = (
+ extras["tokenizers"] + extras["tokenizers"]
+ extras["speech"] + extras["speech"]
+ extras["vision"] + extras["vision"]
+ extras["integrations"]
) )
extras["docs_specific"] = deps_list( extras["docs_specific"] = deps_list(
......
src/transformers/dependency_versions_table.py
View file @ 8d171628
...@@ -29,6 +29,7 @@ deps = { ...@@ -29,6 +29,7 @@ deps = {
"onnxconverter-common": "onnxconverter-common", "onnxconverter-common": "onnxconverter-common",
"onnxruntime-tools": "onnxruntime-tools>=1.4.2", "onnxruntime-tools": "onnxruntime-tools>=1.4.2",
"onnxruntime": "onnxruntime>=1.4.0", "onnxruntime": "onnxruntime>=1.4.0",
"optuna": "optuna",
"packaging": "packaging", "packaging": "packaging",
"parameterized": "parameterized", "parameterized": "parameterized",
"protobuf": "protobuf", "protobuf": "protobuf",
...@@ -38,6 +39,7 @@ deps = { ...@@ -38,6 +39,7 @@ deps = {
"pytest-sugar": "pytest-sugar", "pytest-sugar": "pytest-sugar",
"pytest-xdist": "pytest-xdist", "pytest-xdist": "pytest-xdist",
"python": "python>=3.6.0", "python": "python>=3.6.0",
"ray": "ray",
"recommonmark": "recommonmark", "recommonmark": "recommonmark",
"regex": "regex!=2019.12.17", "regex": "regex!=2019.12.17",
"requests": "requests", "requests": "requests",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment