from abc import ABC, abstractmethod
from typing import Any, Union
from torch_geometric.data import Data
from torch_geometric.resolver import resolver
[docs]class GraphGenerator(ABC):
r"""An abstract base class for generating synthetic graphs."""
@abstractmethod
def __call__(self) -> Data:
r"""To be implemented by :class:`GraphGenerator` subclasses."""
raise NotImplementedError
@staticmethod
def resolve(query: Union[Any, str], *args, **kwargs):
import torch_geometric.datasets.graph_generator as graph_generators
graph_generators = [
gen for gen in vars(graph_generators).values()
if isinstance(gen, type) and issubclass(gen, GraphGenerator)
]
return resolver(graph_generators, {}, query, base_cls=GraphGenerator,
base_cls_repr='Graph', *args, **kwargs)
def __repr__(self) -> str:
return f'{self.__class__.__name__}()'