afonso/core-extra
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

edit some parts where pycharm gives warning on type hint

Browse source
This commit is contained in:
Huy Pham 2020-01-14 14:30:08 -08:00
parent 8a4ad41a47
commit 948a33f201
3 changed files with 7 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

8
daemon/core/gui/nodeutils.py
View file

@ -1,4 +1,4 @@
from typing import TYPE_CHECKING, Set
from typing import TYPE_CHECKING, Optional, Set
from core.api.grpc.core_pb2 import NodeType
from core.gui.images import ImageEnum, Images
@ -14,10 +14,10 @@ class NodeDraw:
def __init__(self):
self.custom: bool = False
self.image = None
self.image_enum: ImageEnum = None
self.image_enum: Optional[ImageEnum] = None
self.image_file = None
self.node_type: core_pb2.NodeType = None
self.model: str = None
self.model: Optional[str] = None
self.services: Set[str] = set()
@classmethod
@ -39,7 +39,7 @@ class NodeDraw:
return node_draw
@classmethod
def from_custom(cls, name: str, image_file: str, services: str):
def from_custom(cls, name: str, image_file: str, services: Set[str]):
node_draw = NodeDraw()
node_draw.custom = True
node_draw.image_file = image_file