1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
from collections import defaultdict
import torch
import torch_geometric.typing
from ._compile import compile, is_compiling
from ._onnx import is_in_onnx_export
from .index import Index
from .edge_index import EdgeIndex
from .seed import seed_everything
from .home import get_home_dir, set_home_dir
from .device import is_mps_available, is_xpu_available, device
from .isinstance import is_torch_instance
from .debug import is_debug_enabled, debug, set_debug
import torch_geometric.utils
import torch_geometric.data
import torch_geometric.sampler
import torch_geometric.loader
import torch_geometric.transforms
import torch_geometric.datasets
import torch_geometric.nn
import torch_geometric.explain
import torch_geometric.profile
from .experimental import (is_experimental_mode_enabled, experimental_mode,
set_experimental_mode)
from .lazy_loader import LazyLoader
contrib = LazyLoader('contrib', globals(), 'torch_geometric.contrib')
graphgym = LazyLoader('graphgym', globals(), 'torch_geometric.graphgym')
__version__ = '2.6.1'
__all__ = [
'Index',
'EdgeIndex',
'seed_everything',
'get_home_dir',
'set_home_dir',
'compile',
'is_compiling',
'is_in_onnx_export',
'is_mps_available',
'is_xpu_available',
'device',
'is_torch_instance',
'is_debug_enabled',
'debug',
'set_debug',
'is_experimental_mode_enabled',
'experimental_mode',
'set_experimental_mode',
'torch_geometric',
'__version__',
]
# Serialization ###############################################################
if torch_geometric.typing.WITH_PT24:
torch.serialization.add_safe_globals([
dict,
list,
defaultdict,
Index,
torch_geometric.index.CatMetadata,
EdgeIndex,
torch_geometric.edge_index.SortOrder,
torch_geometric.edge_index.CatMetadata,
])
|
