make fsspec optional

megvii-research · Nov 20, 2024 · 4d65b59 · 4d65b59
1 parent 20760de
commit 4d65b59
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/megfile/fsspec.py b/megfile/fsspec.py
@@ -3,7 +3,10 @@
 from datetime import datetime
 from typing import IO, BinaryIO, Callable, Iterator, List, Optional, Tuple
 
-import fsspec
+try:
+    import fsspec
+except ImportError:  # pragma: no cover
+    fsspec = None
 
 from megfile.errors import _create_missing_ok_generator
 from megfile.interfaces import ContextIterator, FileEntry, PathLike, StatResult, URIPath
@@ -60,7 +63,7 @@ def _make_entry(filesystem, info):
 
 class BaseFSSpecPath(URIPath):
     protocol: str
-    filesystem: fsspec.AbstractFileSystem
+    filesystem: 'fsspec.AbstractFileSystem'
 
     def __init__(self, path: PathLike, *other_paths: PathLike):
         super().__init__(path, *other_paths)

diff --git a/requirements-fsspec.txt b/requirements-fsspec.txt
@@ -0,0 +1,2 @@
+fsspec
+huggingface_hub