Skip to content

Commit

Permalink
Remove restriction on partition algorithm name
Browse files Browse the repository at this point in the history
Since partition assignment algorithms are independent of DGL, there shouldn't be a restriction on the name of the partition algorithm.

For GraphStorm we have implemented a range partition algorithm, and would like to provide the correct name in the metadata file.
  • Loading branch information
thvasilo committed Apr 26, 2024
1 parent 0de1de3 commit 520e9ee
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions tools/partition_algo/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,8 +75,5 @@ def load_partition_meta(meta_file):
raise DGLError(
f"num_parts[{part_meta.num_parts}] should be greater than 0."
)
if part_meta.algo_name not in ["random", "metis"]:
raise DGLError(
f"algo_name[{part_meta.num_parts}] is not supported."
)

return part_meta

0 comments on commit 520e9ee

Please sign in to comment.