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

Remove DeprecationWarnings from MG Uniform Neighbor Sample test #4722

Open
wants to merge 8 commits into
base: branch-25.02
Choose a base branch
from
Original file line number Diff line number Diff line change
Expand Up @@ -386,7 +386,6 @@ def test_uniform_neighbor_sample_edge_properties(dask_client, return_offsets):
),
fanout_vals=[-1, -1],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=True,
keep_batches_together=True,
min_batch_id=0,
Expand Down Expand Up @@ -503,7 +502,6 @@ def test_uniform_neighbor_sample_edge_properties_self_loops(dask_client):
),
fanout_vals=[2, 2],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=True,
).compute()

Expand Down Expand Up @@ -550,7 +548,6 @@ def test_uniform_neighbor_sample_hop_id_order():
cudf.Series([0, 1], dtype="int64"),
fanout_vals=[2, 2, 2],
with_replacement=False,
with_edge_properties=True,
)

for p in range(sampling_results.npartitions):
Expand Down Expand Up @@ -589,7 +586,6 @@ def test_uniform_neighbor_sample_hop_id_order_multi_batch():
),
fanout_vals=[2, 2, 2],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=True,
)

Expand Down Expand Up @@ -649,7 +645,6 @@ def test_uniform_neighbor_edge_properties_sample_small_start_list(
),
fanout_vals=[10, 25],
with_replacement=with_replacement,
with_edge_properties=True,
with_batch_ids=True,
)

Expand Down Expand Up @@ -687,7 +682,6 @@ def test_uniform_neighbor_sample_without_dask_inputs(dask_client):
),
fanout_vals=[2, 2],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=True,
).compute()

Expand Down Expand Up @@ -753,7 +747,6 @@ def test_uniform_neighbor_sample_batched(dask_client, dataset, input_df, max_bat
start_list=input_vertices,
fanout_vals=[5, 5],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=True,
)

Expand Down Expand Up @@ -799,7 +792,6 @@ def test_uniform_neighbor_sample_exclude_sources_basic(dask_client):
),
[2, 3, 3],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=True,
random_state=62,
prior_sources_behavior="exclude",
Expand Down Expand Up @@ -842,7 +834,6 @@ def test_uniform_neighbor_sample_exclude_sources_email_eu_core(dask_client):
seeds,
[5, 4, 3, 2, 1],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=False,
prior_sources_behavior="exclude",
).compute()
Expand Down Expand Up @@ -888,7 +879,6 @@ def test_uniform_neighbor_sample_carry_over_sources_basic(dask_client):
),
[2, 3, 3],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=True,
random_state=62,
prior_sources_behavior="carryover",
Expand Down Expand Up @@ -936,7 +926,6 @@ def test_uniform_neighbor_sample_carry_over_sources_email_eu_core(dask_client):
seeds,
[5, 4, 3, 2, 1],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=False,
prior_sources_behavior="carryover",
).compute()
Expand Down Expand Up @@ -971,7 +960,6 @@ def test_uniform_neighbor_sample_deduplicate_sources_email_eu_core(dask_client):
seeds,
[5, 4, 3, 2, 1],
with_replacement=False,
with_edge_properties=True,
with_batch_ids=False,
deduplicate_sources=True,
).compute()
Expand Down Expand Up @@ -1007,7 +995,6 @@ def test_uniform_neighbor_sample_renumber(dask_client, hops):
seeds,
hops,
with_replacement=False,
with_edge_properties=True,
with_batch_ids=False,
deduplicate_sources=True,
renumber=True,
Expand Down Expand Up @@ -1050,7 +1037,6 @@ def test_uniform_neighbor_sample_offset_renumber(dask_client, hops):
seeds,
hops,
with_replacement=False,
with_edge_properties=True,
with_batch_ids=False,
deduplicate_sources=True,
renumber=False,
Expand All @@ -1069,7 +1055,6 @@ def test_uniform_neighbor_sample_offset_renumber(dask_client, hops):
seeds,
hops,
with_replacement=False,
with_edge_properties=True,
with_batch_ids=False,
deduplicate_sources=True,
renumber=True,
Expand Down Expand Up @@ -1130,7 +1115,6 @@ def test_uniform_neighbor_sample_csr_csc_global(dask_client, hops, seed):
seeds,
hops,
with_replacement=False,
with_edge_properties=True,
with_batch_ids=False,
deduplicate_sources=True,
# carryover not valid because C++ sorts on (hop,src)
Expand Down Expand Up @@ -1190,7 +1174,6 @@ def test_uniform_neighbor_sample_csr_csc_local(dask_client, hops, seed):
seeds,
hops,
with_replacement=False,
with_edge_properties=True,
with_batch_ids=False,
deduplicate_sources=True,
prior_sources_behavior="carryover",
Expand Down
Loading