Uploads all files in `working_dir` to `repo_id`.
(
self,
working_dir: Union[str, os.PathLike],
repo_id: str,
token: Optional[str] = None,
commit_message: Optional[str] = None,
create_pr: bool = False,
)
| 398 | """ |
| 399 | |
| 400 | def _upload_folder( |
| 401 | self, |
| 402 | working_dir: Union[str, os.PathLike], |
| 403 | repo_id: str, |
| 404 | token: Optional[str] = None, |
| 405 | commit_message: Optional[str] = None, |
| 406 | create_pr: bool = False, |
| 407 | ): |
| 408 | """ |
| 409 | Uploads all files in `working_dir` to `repo_id`. |
| 410 | """ |
| 411 | if commit_message is None: |
| 412 | if "Model" in self.__class__.__name__: |
| 413 | commit_message = "Upload model" |
| 414 | elif "Scheduler" in self.__class__.__name__: |
| 415 | commit_message = "Upload scheduler" |
| 416 | else: |
| 417 | commit_message = f"Upload {self.__class__.__name__}" |
| 418 | |
| 419 | logger.info(f"Uploading the files of {working_dir} to {repo_id}.") |
| 420 | return upload_folder( |
| 421 | repo_id=repo_id, folder_path=working_dir, token=token, commit_message=commit_message, create_pr=create_pr |
| 422 | ) |
| 423 | |
| 424 | def push_to_hub( |
| 425 | self, |
no test coverage detected