Update the types of dataclass attributes according to usage

Fixes #156
wireviz · Oct 23, 2020 · a050e9d · a050e9d
1 parent e2e8bbf
commit a050e9d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/wireviz/DataClasses.py b/src/wireviz/DataClasses.py
@@ -79,8 +79,8 @@ class Connector:
     pinlabels: List[Any] = field(default_factory=list)
     pins: List[Any] = field(default_factory=list)
     color: Optional[str] = None
-    show_name: bool = None
-    show_pincount: bool = None
+    show_name: Optional[bool] = None
+    show_pincount: Optional[bool] = None
     hide_disconnected_pins: bool = False
     autogenerate: bool = False
     loops: List[Any] = field(default_factory=list)
@@ -167,7 +167,7 @@ class Cable:
     length: float = 0
     color: Optional[str] = None
     wirecount: Optional[int] = None
-    shield: bool = False
+    shield: Union[bool, str] = False # False | True | color
     image: Optional[Image] = None
     notes: Optional[str] = None
     colors: List[Any] = field(default_factory=list)