Filter out unknown issues

This commit is contained in:
Jack Grigg 2021-05-07 12:43:13 +12:00
parent 78bedd4f3a
commit 09ef88a54d
1 changed files with 7 additions and 2 deletions

View File

@ -58,9 +58,9 @@ class GitHubIssue:
def __repr__(self):
if self.repo_id in REPOS:
repo = '/'.join(REPOS[self.repo_id])
return '%s#%d' % (repo, self.issue_number)
else:
repo = self.repo_id
return '%s#%d' % (repo, self.issue_number)
return 'Unknown'
def __eq__(self, other):
return (self.repo_id, self.issue_number) == (other.repo_id, other.issue_number)
@ -164,6 +164,11 @@ def main():
# Relable the graph
dg = nx.relabel_nodes(dg, mapping)
# Filter out unknown issues
unknown = [n for n in dg if n.repo_id not in REPOS]
if len(unknown) > 0:
dg.remove_nodes_from(unknown)
# Apply property annotations
for (source, sink) in dg.edges:
attrs = dg.edges[source, sink]