Skip to content

Conversation

@melvinbarbaux
Copy link
Collaborator

Summary

What does this PR change?

Checklist

  • Tests added or updated
  • CI green
  • Changelog updated
  • Documentation updated

Notes

Anything reviewers should know?

@melvinbarbaux melvinbarbaux self-assigned this Jan 29, 2026
@codecov
Copy link

codecov bot commented Jan 29, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

Copy link

@chatgpt-codex-connector chatgpt-codex-connector bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

💡 Codex Review

from torch_geometric.utils import subgraph
# relabel_nodes=True ensures indices map to 0..len(idx)
edge_index, _ = subgraph(idx, X["edge_index"], relabel_nodes=True, num_nodes=n)
batch_data["edge_index"] = edge_index

P1 Badge Normalize slice indices before subgraph

slice_data forwards idx directly into torch_geometric.utils.subgraph, but several call sites pass slice(...) (e.g., ADSH _update_scores and TriNet batching). subgraph expects a tensor/list/mask subset, so a Python slice will raise a TypeError, which means graph dict inputs with edge_index will crash as soon as slice-based batching runs. Converting slices to an index tensor (e.g., torch.arange(start, end, device=...)) before calling subgraph avoids this failure.

ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

Codex can also answer questions or update the PR. Try commenting "@codex address that feedback".

@melvinbarbaux melvinbarbaux merged commit 2ba3da7 into main Jan 29, 2026
8 checks passed
@melvinbarbaux melvinbarbaux deleted the fix/inductive-graph-binary-cleanup branch January 29, 2026 18:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant