[tune/train] Add hard dependency on fsspec ray-project#38581

Signed-off-by: Eric Liang <[email protected]>
miqdigital · Aug 18, 2023 · ca1a546 · ca1a546
1 parent 79bd90a
commit ca1a546
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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",