Faster HfFileSystem.find (#1809)

* Remove `HfFileSystem.exists` override * Add `list_repo_tree` to the API * Rename `BlobLastCommitInfo` to `LastCommitInfo` * Add `get_paths_info` to the API * Improve HfFileSystem tests * Faster `HfFileSystem.find` * Tests * Update src/huggingface_hub/hf_api.py Co-authored-by: Lucain <[email protected]> * Address the rest of comments * Fix comment --------- Co-authored-by: Lucain <[email protected]>
huggingface · Nov 10, 2023 · 7297957 · 7297957
1 parent 18d0ae2
commit 7297957
Show file tree

Hide file tree

Showing 5 changed files with 776 additions and 128 deletions.
diff --git a/src/huggingface_hub/__init__.py b/src/huggingface_hub/__init__.py
@@ -185,6 +185,7 @@
         "get_full_repo_name",
         "get_inference_endpoint",
         "get_model_tags",
+        "get_paths_info",
         "get_repo_discussions",
         "get_space_runtime",
         "get_space_variables",
@@ -200,6 +201,7 @@
         "list_repo_files",
         "list_repo_likers",
         "list_repo_refs",
+        "list_repo_tree",
         "list_spaces",
         "merge_pull_request",
         "model_info",
@@ -519,6 +521,7 @@ def __dir__():
         get_full_repo_name,  # noqa: F401
         get_inference_endpoint,  # noqa: F401
         get_model_tags,  # noqa: F401
+        get_paths_info,  # noqa: F401
         get_repo_discussions,  # noqa: F401
         get_space_runtime,  # noqa: F401
         get_space_variables,  # noqa: F401
@@ -534,6 +537,7 @@ def __dir__():
         list_repo_files,  # noqa: F401
         list_repo_likers,  # noqa: F401
         list_repo_refs,  # noqa: F401
+        list_repo_tree,  # noqa: F401
         list_spaces,  # noqa: F401
         merge_pull_request,  # noqa: F401
         model_info,  # noqa: F401