Sporalize-api / storage_listing.py
Shoraky's picture
Improve backend request performance
82b44e7
Raw History Blame Contribute Delete
728 Bytes
"""Keep metadata requests scoped to the session being viewed."""
from huggingface_hub import RepoFile
from huggingface_hub.errors import RemoteEntryNotFoundError
def session_files(api, repo_id, folder):
try:
return [
entry.path for entry in api.list_repo_tree(
repo_id=repo_id, repo_type="dataset", path_in_repo=folder,
recursive=False, expand=False,
) if isinstance(entry, RepoFile)
]
except RemoteEntryNotFoundError:
# Old sessions can have results but no source video folder.
return []
def session_folder_exists(api, repo_id, folder):
return bool(api.get_paths_info(repo_id=repo_id, repo_type="dataset", paths=[folder]))