diff --git a/polygraph.py b/polygraph.py index 216b113..9ae047a 100755 --- a/polygraph.py +++ b/polygraph.py @@ -20,7 +20,6 @@ def parse_args(): parser.add_argument('--input', '-i', default='network.json', help='Input file in json format') parser.add_argument('--output', '-o', default='network.png', help='Output image file. Format auto-detected from file extension') parser.add_argument('--gender', '-g', action='store_true', help='Indicate gender in output graph') - parser.add_argument('--relationships', '-r', action='store_true', help='Indicate relationship details (std risk, relationship type) in output graph') return parser.parse_args() @@ -50,11 +49,8 @@ def main(): # Add edges to graph edge_style = 'solid' - for (member1,member2,rel,edge) in data['relationships']: - if settings.relationships: - edge_style = edge - relationship_color = relationship_colors[rel] - + for (member1,member2,edge) in data['relationships']: + edge_style = edge graph.add_edge(member1, member2, style=edge_style, color='black') graph.layout()