AutoGL/autogl/module/feature/_generators/__init__.py

from ._basic import OneHotFeatureGenerator
from ._eigen import EigenFeatureGenerator
from ._graphlet import GraphletGenerator
from ._page_rank import PageRankFeatureGenerator
from ._pyg import (
    LocalDegreeProfileGenerator,
    NormalizeFeatures,
    OneHotDegreeGenerator
)

__all__ = [
    "OneHotFeatureGenerator",
    "EigenFeatureGenerator",
    "GraphletGenerator",
    "PageRankFeatureGenerator",
    "LocalDegreeProfileGenerator",
    "NormalizeFeatures",
    "OneHotDegreeGenerator"
]