<filename>cogdl/data/__init__.py
from .data import Graph, Adjacency
from .batch import Batch
from .dataset import Dataset, MultiGraphDataset
from .dataloader import DataLoader

__all__ = ["Graph", "Adjacency", "Batch", "Dataset", "DataLoader", "MultiGraphDataset"]

<filename>cogdl/data/__init__.py
from .data import Graph, Adjacency
from .batch import Batch
from .dataset import Dataset, MultiGraphDataset
from .dataloader import DataLoader

__all__ = ["Graph", "Adjacency", "Batch", "Dataset", "DataLoader", "MultiGraphDataset"]
