From ca1a546ddd769fac2f0dde9847e937dd89bed17e Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Fri, 18 Aug 2023 13:53:31 -0700 Subject: [PATCH] [tune/train] Add hard dependency on fsspec #38581 Signed-off-by: Eric Liang --- python/ray/train/__init__.py | 1 + python/ray/tune/__init__.py | 1 + python/setup.py | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/python/ray/train/__init__.py b/python/ray/train/__init__.py index 84ccf0b28a08..a40ae870acb9 100644 --- a/python/ray/train/__init__.py +++ b/python/ray/train/__init__.py @@ -3,6 +3,7 @@ import pandas # noqa: F401 import requests # noqa: F401 import pyarrow # noqa: F401 + import fsspec # noqa: F401 except ImportError as exc: raise ImportError( "Can't import ray.train as some dependencies are missing. " diff --git a/python/ray/tune/__init__.py b/python/ray/tune/__init__.py index 8ec4cf6b98f7..9c3e565e060d 100644 --- a/python/ray/tune/__init__.py +++ b/python/ray/tune/__init__.py @@ -3,6 +3,7 @@ import pandas # noqa: F401 import requests # noqa: F401 import pyarrow # noqa: F401 + import fsspec # noqa: F401 except ImportError as exc: raise ImportError( "Can't import ray.tune as some dependencies are missing. " diff --git a/python/setup.py b/python/setup.py index a8be879bf855..1102ddd54ef0 100644 --- a/python/setup.py +++ b/python/setup.py @@ -272,7 +272,7 @@ def get_packages(self): else "grpcio", ], "serve": ["uvicorn", "requests", "starlette", "fastapi", "aiorwlock"], - "tune": ["pandas", "tensorboardX>=1.9", "requests", pyarrow_dep], + "tune": ["pandas", "tensorboardX>=1.9", "requests", pyarrow_dep, "fsspec"], "observability": [ "opentelemetry-api", "opentelemetry-sdk",