Skip to content

GitLab

Commit ffc1ca90 authored by SherylHYX's avatar SherylHYX
Browse files

init ScalableGNN

parent b4dec6bb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
torch_geometric_autoscale/__init__.py
View file @ ffc1ca90
...@@ -15,6 +15,7 @@ from .pool import AsyncIOPool # noqa ...@@ -15,6 +15,7 @@ from .pool import AsyncIOPool # noqa
from .metis import metis, permute # noqa from .metis import metis, permute # noqa
from .utils import compute_micro_f1, gen_masks, dropout # noqa from .utils import compute_micro_f1, gen_masks, dropout # noqa
from .loader import SubgraphLoader, EvalSubgraphLoader # noqa from .loader import SubgraphLoader, EvalSubgraphLoader # noqa
from .models import ScalableGNN
__all__ = [ __all__ = [
'get_data', 'get_data',
...@@ -27,5 +28,6 @@ __all__ = [ ...@@ -27,5 +28,6 @@ __all__ = [
'dropout', 'dropout',
'SubgraphLoader', 'SubgraphLoader',
'EvalSubgraphLoader', 'EvalSubgraphLoader',
'ScalableGNN',
'__version__', '__version__',
] ]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment