pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-04 19:44:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

[UnitTest] Fix "[UnitTest] AttributeError: module 'numpy' has no attribute 'ComplexWarning'" in #7766 (#7768)

Browse Source 
Co-authored-by: Stanley Yang <stanyang@amazon.com>
Co-authored-by: Muhammed Fatih BALIN <m.f.balin@gmail.com>
This commit is contained in:
Stanley Yang
2024-09-04 16:33:35 +08:00
committed by GitHub
parent 8fd6c68031
commit 48f47d50a7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/python/common/sampling/test_sampling.py
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.exceptions.ComplexWarning)
g = dgl.graph(([], []), num_nodes=1000).to(F.ctx())
src, dst = dgl.sampling.global_uniform_negative_sampling(
g, 2000, False, True