Skip to content

Commit 76c940b

Browse files
committed
feat: Orchestrator - Launchers - HuggingFace - Using correct API client in storage providers
1 parent f8c1492 commit 76c940b

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

cloud_pipelines_backend/launchers/huggingface_launchers.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,9 @@ def __init__(
4242
self._api_client = client or huggingface_hub.HfApi(token=hf_token)
4343
self._namespace: str = namespace or self._api_client.whoami()["name"]
4444
self._storage_provider = (
45-
huggingface_repo_storage.HuggingFaceRepoStorageProvider()
45+
huggingface_repo_storage.HuggingFaceRepoStorageProvider(
46+
client=self._api_client
47+
)
4648
)
4749
self._job_timeout = job_timeout
4850
self._hf_job_token = hf_job_token
@@ -402,7 +404,9 @@ def get_log(self) -> str:
402404
if self.has_ended:
403405
try:
404406
return (
405-
huggingface_repo_storage.HuggingFaceRepoStorageProvider()
407+
huggingface_repo_storage.HuggingFaceRepoStorageProvider(
408+
client=self._get_api_client()
409+
)
406410
.make_uri(self._log_uri)
407411
.get_reader()
408412
.download_as_text()

0 commit comments

Comments
 (0)