File size: 728 Bytes
82b44e7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
"""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]))