diff --git a/setup.py b/setup.py index 2b74308081e..12347c14a5c 100644 --- a/setup.py +++ b/setup.py @@ -142,6 +142,7 @@ _deps = [ "optimum-benchmark>=0.3.0", "optuna", "optax>=0.0.8,<=0.1.4", + "pandas<2.3.0", # `datasets` requires `pandas` while `pandas==2.3.0` has issues with CircleCI on 2025/06/05 "packaging>=20.0", "parameterized", "phonemizer", @@ -367,7 +368,7 @@ extras["testing"] = ( extras["deepspeed-testing"] = extras["deepspeed"] + extras["testing"] + extras["optuna"] + extras["sentencepiece"] extras["ruff"] = deps_list("ruff") -extras["quality"] = deps_list("datasets", "isort", "ruff", "GitPython", "urllib3", "libcst", "rich") +extras["quality"] = deps_list("datasets", "isort", "ruff", "GitPython", "urllib3", "libcst", "rich", "pandas") extras["all"] = ( extras["tf"] diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 5c0ae6b772f..f5282f9fd39 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -48,6 +48,7 @@ deps = { "optimum-benchmark": "optimum-benchmark>=0.3.0", "optuna": "optuna", "optax": "optax>=0.0.8,<=0.1.4", + "pandas": "pandas<2.3.0", "packaging": "packaging>=20.0", "parameterized": "parameterized", "phonemizer": "phonemizer",