dmlc--dgl
a14f69c97d
* Implemented the following changes. * Remove NUM_NODES_PER_CHUNK * Remove NUM_EDGES_PER_CHUNK * Remove the dependency between no. of edge files per edge type and no. of partitions * Remove the dependency between no. of edge feature files per edge type and no. of partitions * Remove the dependency between no. of edge feature files and no. of edge files per edge type. * Remove the dependency between no. of node feature files and no. of partitions * Add “node_type_counts”. This will be a list of integers. Each integer will represent total count of a node-type. The index in this list and the index in the “node_type” will be the same for a given node-type. * Add “edge_type_counts”. This will be a list of integers. Each integer will represent total count of an edge-type. The index in this list and the index in the “edge_type” list will be the same for a given edge-type. * Applying lintrunner patch. * Adding missing keys to the metadata in the unit test framework. * lintrunner patch. * Resolving CI test failures due to merge conflicts. * Applying lintrunner patch * applying lintrunner patch * Replacing tabspace with spaces - to satisfy lintrunner * Fixing the CI Test Failure cases. * Applying lintrunner patch * lintrunner complaining about a blank line. * Resolving issues with print statement for NoneType * Removed tests for the arbitrary chunks tests. Since this functionality is not supported anymore. * Addressing CI review comments. * addressing CI review comments * lintrunner patch * lintrunner patch. * Addressing CI review comments. * lintrunner patch.
44 行
1022 B
Python
44 行
1022 B
Python
GLOBAL_NID = "global_node_id"
|
|
GLOBAL_EID = "global_edge_id"
|
|
|
|
SHUFFLE_GLOBAL_NID = "shuffle_global_node_id"
|
|
SHUFFLE_GLOBAL_EID = "shuffle_global_edge_id"
|
|
|
|
NTYPE_ID = "node_type_id"
|
|
ETYPE_ID = "edge_type_id"
|
|
|
|
GLOBAL_TYPE_NID = "global_type_node_id"
|
|
GLOBAL_TYPE_EID = "global_type_edge_id"
|
|
|
|
GLOBAL_SRC_ID = "global_src_id"
|
|
GLOBAL_DST_ID = "global_dst_id"
|
|
SHUFFLE_GLOBAL_SRC_ID = "shuffle_global_src_id"
|
|
SHUFFLE_GLOBAL_DST_ID = "shuffle_global_dst_id"
|
|
|
|
OWNER_PROCESS = "owner_proc_id"
|
|
|
|
PART_LOCAL_NID = "part_local_nid"
|
|
|
|
STR_NODE_TYPE = "node_type"
|
|
STR_EDGE_TYPE = "edge_type"
|
|
STR_EDGES = "edges"
|
|
STR_FORMAT = "format"
|
|
STR_FORMAT_DELIMITER = "delimiter"
|
|
STR_DATA = "data"
|
|
STR_NODE_DATA = "node_data"
|
|
STR_EDGE_DATA = "edge_data"
|
|
|
|
STR_NUMPY = "numpy"
|
|
STR_PARQUET = "parquet"
|
|
STR_CSV = "csv"
|
|
STR_NAME = "name"
|
|
|
|
STR_GRAPH_NAME = "graph_name"
|
|
STR_NODE_FEATURES = "node_features"
|
|
STR_EDGE_FEATURES = "edge_features"
|
|
|
|
STR_NUM_NODES_PER_TYPE = "num_nodes_per_type"
|
|
STR_NUM_EDGES_PER_TYPE = "num_edges_per_type"
|
|
|
|
STR_NTYPES = "ntypes"
|