AutoGL/autogl/datasets/utils/__init__.py

from ._split_edges import split_edges
from ._general import (
    index_to_mask,
    random_splits_mask,
    random_splits_mask_class,
    graph_cross_validation,
    graph_random_splits,
    graph_get_split,
    set_fold,
)