| from functools import partial | |
| from huggingface_hub import hf_hub_url | |
| from huggingface_hub.utils import get_session, hf_raise_for_status | |
| hf_dataset_url = partial(hf_hub_url, repo_type="dataset") | |
| def check_auth(hf_api, repo_id, token=None): | |
| headers = hf_api._build_hf_headers(token=token) | |
| path = f"{hf_api.endpoint}/api/datasets/{repo_id}/auth-check" | |
| r = get_session().get(path, headers=headers) | |
| hf_raise_for_status(r) | |