Skip to content

Commit

Permalink
revert change to dgl sampler
Browse files Browse the repository at this point in the history
  • Loading branch information
alexbarghi-nv committed Aug 28, 2024
1 parent 4571711 commit 3131926
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def sample(

if g.is_homogeneous:
indices = torch.concat(list(indices))
reader = ds.sample_from_nodes(indices, batch_size=batch_size)
reader = ds.sample_from_nodes(indices.long(), batch_size=batch_size)
return HomogeneousSampleReader(reader, self.output_format, self.edge_dir)

raise ValueError(
Expand Down

0 comments on commit 3131926

Please sign in to comment.