Source code for analogvnn.graph.GraphEnum

from __future__ import annotations

import enum
from typing import Union, Callable

__all__ = ['GraphEnum', 'GRAPH_NODE_TYPE']


[docs]class GraphEnum(enum.Enum):
    """The graph enum for indicating input, output and stop.

    Attributes:
        INPUT (GraphEnum): GraphEnum.INPUT
        OUTPUT (GraphEnum): GraphEnum.OUTPUT
        STOP (GraphEnum): GraphEnum.STOP
    """

[docs]    INPUT = 'INPUT'
[docs]    OUTPUT = 'OUTPUT'
[docs]    STOP = 'STOP'


[docs]GRAPH_NODE_TYPE = Union[GraphEnum, Callable]