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
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",