Guarantee insert_finished() on bulk AllGather

[Matt711]

Description

Need to make sure insert_finished is called for collective operations.

Checklist

• I am familiar with the Contributing Guidelines.
• New or existing tests cover these changes.
• The documentation is up to date with these changes.

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: 4023b4c2-f460-4ef4-9095-a38d6637fdaf

📥 Commits

Reviewing files that changed from the base of the PR and between 882878f and a4b6e2f.

📒 Files selected for processing (1)

• python/cudf_polars/cudf_polars/experimental/rapidsmpf/frontend/core.py

🚧 Files skipped from review as they are similar to previous changes (1)

• python/cudf_polars/cudf_polars/experimental/rapidsmpf/frontend/core.py

---

📝 Walkthrough

Summary by CodeRabbit

• Bug Fixes
  • Enhanced reliability of data gathering operations by ensuring cleanup procedures are properly executed even when errors occur during data processing.

Walkthrough

The all_gather_host_data function wraps buffer insertion operations in a try/finally block to guarantee that AllGather.insert_finished() is called even if data packing or insertion raises an exception. A TODO comment documents a future refactor to make AllGather itself a context manager.

Changes

AllGather Error Handling

Layer / File(s)	Summary
AllGather insertion with guaranteed cleanup python/cudf_polars/cudf_polars/experimental/rapidsmpf/frontend/core.py	all_gather_host_data wraps PackedData.from_host_bytes(...) and allgather.insert(...) in a try/finally block to ensure allgather.insert_finished() executes even if those calls raise, with a TODO noting that AllGather should become a context manager.

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~8 minutes

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title directly and clearly describes the main change: ensuring insert_finished() is called on bulk AllGather operations, which aligns with the changeset's primary objective.
Description check	✅ Passed	The description is related to the changeset, stating the need to ensure insert_finished is called for collective operations, which matches the primary change in the code.
Docstring Coverage	✅ Passed	Docstring coverage is 100.00% which is sufficient. The required threshold is 80.00%.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests

---

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

[Matt711]

/merge

[madsbk]

Good catch!

[mroeschke]

Probably good to retarget to release/26.06

[rapids-bot]

This PR's base branch has been changed since the last /merge command. Please issue the command again to confirm the merge with the new base branch.

[Matt711]

/merge