Skip to content

Commit 9a5dc6b

Browse files
authored
set library name in HfApi
1 parent 6d66cab commit 9a5dc6b

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
@@ -150,7 +150,7 @@ def __init__(
150150
def _get_api(self):
151151
from huggingface_hub import HfApi
152152

153-
return HfApi(token=self.token, endpoint=self.endpoint)
153+
return HfApi(token=self.token, endpoint=self.endpoint, library_name="pyspark_huggingface")
154154

155155
@property
156156
def prefix(self) -> str:

0 commit comments

Comments
 (0)