From b2be5608072992c7ef99b8fb3953c69510a43871 Mon Sep 17 00:00:00 2001 From: Xyphuz Date: Wed, 8 Jun 2022 23:29:01 +0800 Subject: [PATCH] chore: add missing type annotations --- compose_viz/graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/compose_viz/graph.py b/compose_viz/graph.py index 83eed72..ea27581 100644 --- a/compose_viz/graph.py +++ b/compose_viz/graph.py @@ -5,7 +5,7 @@ import graphviz from compose_viz.models.compose import Compose -def apply_vertex_style(type) -> dict: +def apply_vertex_style(type: str) -> dict: style = { "service": { "shape": "component", @@ -36,7 +36,7 @@ def apply_vertex_style(type) -> dict: return style[type] -def apply_edge_style(type) -> dict: +def apply_edge_style(type: str) -> dict: style = { "exposes": { "style": "solid",