chore: add missing type annotations

This commit is contained in:
Xyphuz 2022-06-08 23:29:01 +08:00
parent e0ef69febd
commit b2be560807

View file

@ -5,7 +5,7 @@ import graphviz
from compose_viz.models.compose import Compose from compose_viz.models.compose import Compose
def apply_vertex_style(type) -> dict: def apply_vertex_style(type: str) -> dict:
style = { style = {
"service": { "service": {
"shape": "component", "shape": "component",
@ -36,7 +36,7 @@ def apply_vertex_style(type) -> dict:
return style[type] return style[type]
def apply_edge_style(type) -> dict: def apply_edge_style(type: str) -> dict:
style = { style = {
"exposes": { "exposes": {
"style": "solid", "style": "solid",