Skip to content

Extract sampler helpers to utils module and add shared-backend message types#573

Closed
kmontemayor2-sc wants to merge 1 commit intomainfrom
kmonte/shared-backend-pr1-helpers
Closed

Extract sampler helpers to utils module and add shared-backend message types#573
kmontemayor2-sc wants to merge 1 commit intomainfrom
kmonte/shared-backend-pr1-helpers

Conversation

@kmontemayor2-sc
Copy link
Copy Markdown
Collaborator

…e types

Move _create_dist_sampler() and _prepare_degree_tensors() from dist_sampling_producer.py into a new gigl/distributed/utils/dist_sampler.py module as public functions, along with SamplerInput and SamplerRuntime type aliases. Update dist_sampling_producer.py to import from the new module.

Add InitSamplingBackendRequest and RegisterBackendRequest frozen dataclasses to messages.py for the upcoming shared sampling backend.

Scope of work done

Where is the documentation for this feature?: N/A

Did you add automated tests or write a test plan?

Updated Changelog.md? NO

Ready for code review?: NO

@kmontemayor2-sc
Copy link
Copy Markdown
Collaborator Author

/all_test

@github-actions
Copy link
Copy Markdown
Contributor

github-actions bot commented Apr 6, 2026

GiGL Automation

@ 17:19:08UTC : 🔄 E2E Test started.

@ 18:50:45UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions bot commented Apr 6, 2026

GiGL Automation

@ 17:19:09UTC : 🔄 Integration Test started.

@ 18:48:37UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions bot commented Apr 6, 2026

GiGL Automation

@ 17:19:11UTC : 🔄 Lint Test started.

@ 17:26:36UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions bot commented Apr 6, 2026

GiGL Automation

@ 17:19:12UTC : 🔄 Scala Unit Test started.

@ 17:27:33UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Copy Markdown
Contributor

github-actions bot commented Apr 6, 2026

GiGL Automation

@ 17:19:12UTC : 🔄 Python Unit Test started.

@ 18:31:44UTC : ✅ Workflow completed successfully.

…e types

Move _create_dist_sampler() and _prepare_degree_tensors() from
dist_sampling_producer.py into a new gigl/distributed/utils/dist_sampler.py
module as public functions, along with SamplerInput and SamplerRuntime type
aliases. Update dist_sampling_producer.py to import from the new module.

Add InitSamplingBackendRequest and RegisterBackendRequest frozen dataclasses
to messages.py for the upcoming shared sampling backend.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@kmontemayor2-sc kmontemayor2-sc force-pushed the kmonte/shared-backend-pr1-helpers branch from b77cdaa to 3b0965c Compare April 6, 2026 17:28
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.

2 participants