Uploads all files in `working_dir` to `repo_id`.
(
self,
working_dir: str | os.PathLike,
repo_id: str,
token: str | None = None,
commit_message: str | None = None,
create_pr: bool = False,
subfolder: str | None = None,
)
| 486 | """ |
| 487 | |
| 488 | def _upload_folder( |
| 489 | self, |
| 490 | working_dir: str | os.PathLike, |
| 491 | repo_id: str, |
| 492 | token: str | None = None, |
| 493 | commit_message: str | None = None, |
| 494 | create_pr: bool = False, |
| 495 | subfolder: str | None = None, |
| 496 | ): |
| 497 | """ |
| 498 | Uploads all files in `working_dir` to `repo_id`. |
| 499 | """ |
| 500 | if commit_message is None: |
| 501 | if "Model" in self.__class__.__name__: |
| 502 | commit_message = "Upload model" |
| 503 | elif "Scheduler" in self.__class__.__name__: |
| 504 | commit_message = "Upload scheduler" |
| 505 | else: |
| 506 | commit_message = f"Upload {self.__class__.__name__}" |
| 507 | |
| 508 | logger.info(f"Uploading the files of {working_dir} to {repo_id}.") |
| 509 | return upload_folder( |
| 510 | repo_id=repo_id, |
| 511 | folder_path=working_dir, |
| 512 | token=token, |
| 513 | commit_message=commit_message, |
| 514 | create_pr=create_pr, |
| 515 | path_in_repo=subfolder, |
| 516 | ) |
| 517 | |
| 518 | def push_to_hub( |
| 519 | self, |
no test coverage detected