diff --git a/python/setup.py b/python/setup.py
index 869d5dfabecf..6723ec78554c 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -279,6 +279,8 @@ def get_packages(self):
         ],
         "tune": [
             "pandas",
+            # TODO: Remove pydantic dependency from tune once tune doesn't import train
+            pydantic_dep,
             "tensorboardX>=1.9",
             "requests",
             *pyarrow_deps,