Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[distGB] fix local variable 'sorted_idx' referenced before assignment in convert_partition.py #7830

Open
wants to merge 15 commits into
base: master
Choose a base branch
from
Prev Previous commit
Next Next commit
change format
  • Loading branch information
Ubuntu committed Oct 21, 2024
commit a0b020c6a08f71df6949f95685c4c89f1c3c6e44
7 changes: 4 additions & 3 deletions tests/tools/test_dist_part.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,8 @@

from distpartitioning import array_readwriter
from distpartitioning.utils import generate_read_list
from pytest_utils import create_chunked_dataset
from pytest_utils import chunk_graph, create_hetero_chunked_dataset
from scipy import sparse as spsp

from tools.verification_utils import (
verify_graph_feats,
Expand All @@ -39,7 +40,7 @@ def _test_chunk_graph(
num_chunks_edge_data=None,
):
with tempfile.TemporaryDirectory() as root_dir:
g = create_chunked_dataset(
g = create_hetero_chunked_dataset(
root_dir,
num_chunks,
data_fmt=data_fmt,
Expand Down Expand Up @@ -220,7 +221,7 @@ def _test_pipeline(
return

with tempfile.TemporaryDirectory() as root_dir:
g = create_chunked_dataset(
g = create_hetero_chunked_dataset(
root_dir,
num_chunks,
data_fmt=data_fmt,
Expand Down