nx-neptune/nx_neptune/__init__.py at 01dfdb9c9b7fc0d3b3d00229bff536dc0d1752e7 · awslabs/nx-neptune · GitHub

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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
# Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"). You
# may not use this file except in compliance with the License. A copy of
# the License is located at
#
#     http://aws.amazon.com/apache2.0/
#
# or in the "license" file accompanying this file. This file is
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
# ANY KIND, either express or implied. See the License for the specific
# language governing permissions and limitations under the License.
from .algorithms import louvain_communities
from .algorithms.centrality.closeness import closeness_centrality
from .algorithms.centrality.degree_centrality import (
    degree_centrality,
    in_degree_centrality,
    out_degree_centrality,
)
from .algorithms.communities.label_propagation import (
    asyn_lpa_communities,
    fast_label_propagation_communities,
    label_propagation_communities,
)
from .algorithms.link_analysis.pagerank import pagerank
from .algorithms.traversal.bfs import bfs_edges, bfs_layers, descendants_at_distance
from .clients import Edge, Node
from .instance_management import (
    create_csv_table_from_s3,
    create_graph_snapshot,
    create_iceberg_table_from_table,
    create_na_instance,
    create_na_instance_from_snapshot,
    create_na_instance_with_s3_import,
    delete_graph_snapshot,
    delete_na_instance,
    drop_athena_table,
    empty_s3_bucket,
    export_athena_table_to_s3,
    export_csv_to_s3,
    get_athena_query_results,
    import_csv_from_s3,
    start_na_instance,
    stop_na_instance,
    update_na_instance_size,
    validate_athena_query,
    validate_permissions,
)
from .interface import BackendInterface
from .na_graph import (
    NETWORKX_GRAPH_ID,
    NETWORKX_S3_IAM_ROLE_ARN,
    NeptuneGraph,
    set_config_graph_id,
)
from .session_manager import CleanupTask, SessionManager
from .utils.decorators import configure_if_nx_active

__version__ = "0.4.4"

__all__ = [
    # environment variables
    "NETWORKX_GRAPH_ID",
    "NETWORKX_S3_IAM_ROLE_ARN",
    # algorithms
    "bfs_edges",
    "bfs_layers",
    "descendants_at_distance",
    "pagerank",
    "degree_centrality",
    "in_degree_centrality",
    "out_degree_centrality",
    "label_propagation_communities",
    "asyn_lpa_communities",
    "fast_label_propagation_communities",
    "louvain_communities",
    # graphs
    "Node",
    "Edge",
    "NeptuneGraph",
    "set_config_graph_id",
    # decorators
    "configure_if_nx_active",
    "BackendInterface",
    # instance management
    "validate_permissions",
    "create_na_instance",
    "create_na_instance_with_s3_import",
    "create_na_instance_from_snapshot",
    "delete_na_instance",
    "update_na_instance_size",
    "start_na_instance",
    "stop_na_instance",
    "create_graph_snapshot",
    "delete_graph_snapshot",
    "import_csv_from_s3",
    "export_csv_to_s3",
    "empty_s3_bucket",
    "validate_athena_query",
    "export_athena_table_to_s3",
    "create_csv_table_from_s3",
    "create_iceberg_table_from_table",
    "drop_athena_table",
    "get_athena_query_results",
    # session management
    "SessionManager",
    "CleanupTask",
]