Method check_edge

python/graphscope/learning/gl_torch_graph.py:59–66

(schema, edge)

    @staticmethod
    def check_edge(schema, edge):
        if not isinstance(edge, tuple) or len(edge) != 3:
            raise ValueError("Each edge should be a tuple of length 3")
        for vertex_label in [edge[0], edge[2]]:
            if vertex_label not in schema.vertex_labels:
                raise ValueError(f"Invalid edge label: {vertex_label}")
        if edge[1] not in schema.edge_labels:
            raise ValueError(f"Invalid edge label: {edge[1]}")

    @staticmethod
    def check_edges(schema, edges):