dmlc--dgl
Minjie Wang
f13b9b6212
* WIP: api * dgl.sampling, dgl.data * dgl.sampling; dgl.dataloading * sampling packages * convert * subgraph * deprecate * subgraph APIs * All docstrings for convert/subgraph/transform * almost all funcs under dgl namespace * WIP: DGLGraph * done graph query * message passing functions * lint * fix merge error * fix test * lint * fix Co-authored-by: Quan Gan <coin2028@hotmail.com>
45 行
1.3 KiB
Python
45 行
1.3 KiB
Python
"""
|
|
The ``dgl`` package contains data structure for storing structural and feature data
|
|
(i.e., the :class:`DGLGraph` class) and also utilities for generating, manipulating
|
|
and transforming graphs.
|
|
"""
|
|
|
|
|
|
# Windows compatibility
|
|
# This initializes Winsock and performs cleanup at termination as required
|
|
import socket
|
|
|
|
# Need to ensure that the backend framework is imported before load dgl libs,
|
|
# otherwise weird cuda problem happens
|
|
from .backend import load_backend, backend_name
|
|
|
|
from . import function
|
|
from . import contrib
|
|
from . import container
|
|
from . import distributed
|
|
from . import random
|
|
from . import sampling
|
|
from . import dataloading
|
|
from . import ops
|
|
|
|
from ._ffi.runtime_ctypes import TypeCode
|
|
from ._ffi.function import register_func, get_global_func, list_global_func_names, extract_ext_funcs
|
|
from ._ffi.base import DGLError, __version__
|
|
|
|
from .base import ALL, NTYPE, NID, ETYPE, EID
|
|
from .readout import *
|
|
from .batch import *
|
|
from .convert import *
|
|
from .generators import *
|
|
from .heterograph import DGLHeteroGraph
|
|
from .heterograph import DGLHeteroGraph as DGLGraph # pylint: disable=reimported
|
|
from .subgraph import *
|
|
from .traversal import *
|
|
from .transform import *
|
|
from .propagate import *
|
|
from .random import *
|
|
from .data.utils import save_graphs, load_graphs
|
|
|
|
from ._deprecate.graph import DGLGraph as DGLGraphStale
|
|
from ._deprecate.nodeflow import *
|