Source code for gigl.src.common.types.exception

from enum import Enum



[docs]
class SubgraphSamplingValidationErrorType(Enum):

[docs]
    REPEATED_OP_NAME = "REPEATED_OP_NAME"


[docs]
    BAD_INPUT_OP_NAME = "BAD_INPUT_OP_NAME"


[docs]
    DAG_CONTAINS_CYCLE = "DAG_CONTAINS_CYCLE"


[docs]
    REPEATED_ROOT_NODE_TYPE = "REPEATED_ROOT_NODE_TYPE"


[docs]
    ROOT_NODE_TYPE_NOT_IN_GRAPH_METADATA = "ROOT_NODE_TYPE_NOT_IN_GRAPH_METADATA"


[docs]
    ROOT_NODE_TYPE_NOT_IN_TASK_METADATA = "ROOT_NODE_TYPE_NOT_IN_TASK_METADATA"


[docs]
    MISSING_ROOT_SAMPLING_OP = "MISSING_ROOT_SAMPLING_OP"


[docs]
    SAMPLING_OP_EDGE_TYPE_NOT_IN_GRAPH_METADATA = (
        "SAMPLING_OP_EDGE_TYPE_NOT_IN_GRAPH_METADATA"
    )


[docs]
    MISSING_EXPECTED_ROOT_NODE_TYPE = "MISSING_EXPECTED_ROOT_NODE_TYPE"


[docs]
    CONTAINS_INVALID_EDGE_IN_DAG = "CONTAINS_INVALID_EDGE_IN_DAG"





[docs]
class SubgraphSamplingValidationError(Exception):
    def __init__(self, message: str, error_type: SubgraphSamplingValidationErrorType):
        super().__init__(message)

[docs]
        self.message = message


[docs]
        self.error_type = error_type


    def __str__(self):
        return f"{self.message} (Error Type: {self.error_type.value})"