Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
rusty1s committed Sep 15, 2023
1 parent 9d93cde commit 0607d72
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion test/csrc/sampler/test_dist_relabel.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -230,7 +230,8 @@ TEST(DistHeteroDisjointRelabelNeighborhoodTest, BasicAssertions) {
/*sampled_neighbors_per_node=*/sampled_neighbors_per_node_dict,
/*num_nodes_dict=*/num_nodes_dict,
/*batch_dict=*/batch_dict,
/*csc=*/false, /*disjoint=*/true);
/*csc=*/false,
/*disjoint=*/true);

auto expected_row = at::tensor({0, 0, 1, 1}, options);
EXPECT_TRUE(at::equal(std::get<0>(relabel_out).at(rel_key), expected_row));
Expand Down

0 comments on commit 0607d72

Please sign in to comment.