diff --git a/setup.py b/setup.py index d3251af1f7c..29d516991df 100644 --- a/setup.py +++ b/setup.py @@ -50,8 +50,6 @@ "numpy<2.0", ] -MYSQL_REQUIRES = ["SQLAlchemy==1.4.17"] - NOTEBOOK_REQUIRES = ["jupyter"] UNITTEST_MINIMAL_REQUIRES = [ @@ -60,9 +58,7 @@ "torchx", # For torchx unit tests. ] -UNITTEST_REQUIRES = ( - DEV_REQUIRES + MYSQL_REQUIRES + NOTEBOOK_REQUIRES + UNITTEST_MINIMAL_REQUIRES -) +UNITTEST_REQUIRES = DEV_REQUIRES + NOTEBOOK_REQUIRES + UNITTEST_MINIMAL_REQUIRES TUTORIAL_REQUIRES = UNITTEST_REQUIRES + [ "ray", # Required for building RayTune tutorial notebook. @@ -114,7 +110,6 @@ def setup_package() -> None: }, extras_require={ "dev": DEV_REQUIRES, - "mysql": MYSQL_REQUIRES, "notebook": NOTEBOOK_REQUIRES, "unittest": UNITTEST_REQUIRES, "unittest_minimal": UNITTEST_MINIMAL_REQUIRES,