File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
cloud_pipelines_backend/launchers Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments