Skip to content

Commit 9a3a91a

Browse files
committed
fix import for compatibility with older huggingface API
1 parent 580864e commit 9a3a91a

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)