AutoGL/autogl/solver/classifier/__init__.py

"""
Auto classifier for classification problems.
"""

from .base import BaseClassifier
from .graph_classifier import AutoGraphClassifier
from .node_classifier import AutoNodeClassifier
from .link_predictor import AutoLinkPredictor
from autogl.backend import DependentBackend
if DependentBackend.is_dgl():
    from .hetero import AutoHeteroNodeClassifier
if DependentBackend.is_pyg():
    from .ssl import SSLGraphClassifier

__all__ = [
    "BaseClassifier",
    "AutoGraphClassifier",
    "AutoNodeClassifier",
    "AutoLinkPredictor",
]

if DependentBackend.is_dgl():
    __all__.extend(['AutoHeteroNodeClassifier'])

if DependentBackend.is_pyg():
    __all__.extend(['SSLGraphClassifier'])