import dataclasses from enum import Enum class NodeType(Enum): INPUT = 1 HIDDEN = 2 OUTPUT = 3 @dataclasses.dataclass(frozen=True) class NodeGene: id: int type: NodeType