Skip to content

Commit 6d66cab

Browse files
authored
Merge pull request #9 from huggingface/haoyu-fix-import-error
Fix import for compatibility with older huggingface_hub
2 parents 580864e + 9a3a91a commit 6d66cab

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyspark_huggingface/huggingface_sink.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def _list_split(self, api: "HfApi") -> Iterator[Union["RepoFile", "RepoFolder"]]
160160
"""
161161
Get all existing files of the current split.
162162
"""
163-
from huggingface_hub.errors import EntryNotFoundError
163+
from huggingface_hub.utils import EntryNotFoundError
164164

165165
try:
166166
objects = api.list_repo_tree(

0 commit comments

Comments
 (0)