diff --git a/test/csrc/sampler/test_neighbor.cpp b/test/csrc/sampler/test_neighbor.cpp index 49baaab58..c7091fc48 100644 --- a/test/csrc/sampler/test_neighbor.cpp +++ b/test/csrc/sampler/test_neighbor.cpp @@ -233,7 +233,7 @@ TEST(BiasedNeighborTest, BasicAssertions) { /*num_neighbors=*/{1}, /*node_time=*/c10::nullopt, /*seed_time=*/c10::nullopt, - /*edge_time=*/edge_weight); + /*edge_time=*/edge_weight, /*edge_weight=*/edge_weight); auto expected_row = at::tensor({0, 1}, options);