This commit is contained in:
Stanley Yang
2024-09-03 08:35:42 +00:00
parent 9e6602f698
commit b4df206c64

View File

@@ -1774,7 +1774,7 @@ def test_sample_neighbors_exclude_edges_homoG(dtype, fused):
@pytest.mark.parametrize("dtype", ["int32", "int64"])
def test_global_uniform_negative_sampling(dtype):
warnings.simplefilter("ignore", np.ComplexWarning)
# warnings.simplefilter("ignore", np.ComplexWarning)
g = dgl.graph(([], []), num_nodes=1000).to(F.ctx())
src, dst = dgl.sampling.global_uniform_negative_sampling(
g, 2000, False, True