diff --git a/studio/backend/plugins/data-designer-github-repo-seed/src/data_designer_github_repo_seed/scraper.py b/studio/backend/plugins/data-designer-github-repo-seed/src/data_designer_github_repo_seed/scraper.py index 6aaeefbe00..3c60cddaf9 100644 --- a/studio/backend/plugins/data-designer-github-repo-seed/src/data_designer_github_repo_seed/scraper.py +++ b/studio/backend/plugins/data-designer-github-repo-seed/src/data_designer_github_repo_seed/scraper.py @@ -189,7 +189,11 @@ def scrape(cfg: ScrapeConfig, base_dir: Path): default_branch = ( default_ref.get("name") if isinstance(default_ref, dict) else None ) - branch = f"refs/heads/{default_branch}" if default_branch else "refs/heads/main" + branch = ( + f"refs/heads/{default_branch}" + if default_branch + else "refs/heads/main" + ) scraper.scrape_commits(branch = branch) finally: scraper.close() diff --git a/studio/backend/plugins/data-designer-github-repo-seed/src/data_designer_github_repo_seed/scraper_impl/scraper.py b/studio/backend/plugins/data-designer-github-repo-seed/src/data_designer_github_repo_seed/scraper_impl/scraper.py index eaeff47da2..19973bdfe1 100644 --- a/studio/backend/plugins/data-designer-github-repo-seed/src/data_designer_github_repo_seed/scraper_impl/scraper.py +++ b/studio/backend/plugins/data-designer-github-repo-seed/src/data_designer_github_repo_seed/scraper_impl/scraper.py @@ -693,9 +693,15 @@ def main(): if not only or "commits" in only: default_ref = repo_meta.get("defaultBranchRef") or {} default_branch = ( - default_ref.get("name") if isinstance(default_ref, dict) else None + default_ref.get("name") + if isinstance(default_ref, dict) + else None + ) + branch = ( + f"refs/heads/{default_branch}" + if default_branch + else "refs/heads/main" ) - branch = f"refs/heads/{default_branch}" if default_branch else "refs/heads/main" scraper.scrape_commits(branch = branch) finally: scraper.close()