Adding pandas optional deps for TFDS dataset compatibility

PiperOrigin-RevId: 687355649
PAIR-code · Oct 18, 2024 · 0c68496 · 0c68496
1 parent 943270a
commit 0c68496
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,7 +25,7 @@ dependencies = [
   "matplotlib>=3.7.1",
   "ml-collections>=0.1.1",
   "numpy>=1.24.1,<2.0.0",
-  "pandas>=2.0.3",
+  "pandas[output-formatting]>=2.0.3",
   "Pillow>=10.0.0",
   "portpicker>=1.5.2",
   "requests>=2.31.0",

diff --git a/requirements.txt b/requirements.txt
@@ -24,7 +24,7 @@ Levenshtein>=0.21.1
 matplotlib>=3.6.0,<3.9.0
 ml-collections>=0.1.1
 numpy>=1.24.1,<2.0.0
-pandas>=2.0.3
+pandas[output-formatting]>=2.0.3
 Pillow>=10.0.0
 portpicker>=1.5.2
 requests>=2.31.0