diff --git a/setup.py b/setup.py index abafc03bc6..d8f58360d7 100644 --- a/setup.py +++ b/setup.py @@ -185,6 +185,7 @@ _deps = [ "tqdm>=4.27", "unidic>=1.0.2", "unidic_lite>=1.0.7", + "urllib3<2.0.0", "uvicorn", ] @@ -331,7 +332,7 @@ extras["testing"] = ( extras["deepspeed-testing"] = extras["deepspeed"] + extras["testing"] + extras["optuna"] + extras["sentencepiece"] -extras["quality"] = deps_list("black", "datasets", "isort", "ruff", "GitPython", "hf-doc-builder") +extras["quality"] = deps_list("black", "datasets", "isort", "ruff", "GitPython", "hf-doc-builder", "urllib3") extras["all"] = ( extras["tf"] diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index bae19acd3e..dd1055ebd2 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -85,5 +85,6 @@ deps = { "tqdm": "tqdm>=4.27", "unidic": "unidic>=1.0.2", "unidic_lite": "unidic_lite>=1.0.7", + "urllib3": "urllib3<2.0.0", "uvicorn": "uvicorn", }