Skip and log missing branches for release_data.py

This commit is contained in:
Sam Wu
2024-04-25 09:48:37 -06:00
parent 2258d75967
commit 25065ec999

View File

@@ -4,6 +4,7 @@ from dataclasses import dataclass, field
import os
import re
import shutil
import sys
from typing import Optional, Union, Dict, List, Tuple
from github import Github, UnknownObjectException
from github.Repository import Repository
@@ -352,6 +353,8 @@ class ReleaseBundleFactory:
"""Create a release bundle of libraries."""
tag_name = f"rocm-{version}"
libraries = { }
missing_branches = []
print(f"\nLibraries for rocm-{version}:")
for name, remote in names_and_remotes:
@@ -365,7 +368,13 @@ class ReleaseBundleFactory:
continue
print(f" Defaulting to branch: {self.branch}")
commit = repo.get_branch(self.branch).commit.sha
try:
repo_branch = repo.get_branch(self.branch)
commit = repo_branch.commit.sha
except Exception:
print(f" - Could not find branch : {self.branch}")
missing_branches.append(f"{self.branch} for {name}")
continue
libraries[name] = ReleaseLib(
name=name,
@@ -381,6 +390,9 @@ class ReleaseBundleFactory:
libraries=libraries
)
for missing in missing_branches:
print(f"Could not find the following branch: {missing}")
return data
def create_data_dict(