From 09ef88a54df18dfe171dd860aeee19d2375e9565 Mon Sep 17 00:00:00 2001 From: Jack Grigg Date: Fri, 7 May 2021 12:43:13 +1200 Subject: [PATCH] Filter out unknown issues --- zcash-core-graph.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/zcash-core-graph.py b/zcash-core-graph.py index 6d5cda64..0ee60c64 100755 --- a/zcash-core-graph.py +++ b/zcash-core-graph.py @@ -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]