Skip to content

Update ABLP Loader in preparation for graph store#478

Draft
kmontemayor2-sc wants to merge 4 commits intomainfrom
kmonte/setup-ablp-loader-for-gs
Draft

Update ABLP Loader in preparation for graph store#478
kmontemayor2-sc wants to merge 4 commits intomainfrom
kmonte/setup-ablp-loader-for-gs

Conversation

@kmontemayor2-sc
Copy link
Collaborator

@kmontemayor2-sc kmontemayor2-sc commented Feb 5, 2026

Scope of work done

Breaking out the changes to support colocated in graph store mode to their own PR (this one) so it's easier to review. In hindsight, #432 was probably too complicated and the ABLP changes will be even more so.

See #475 for the MVP here.

This PR should be a no-op (other than fixing some typos in logs).

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
Collaborator Author

/unit_test_py

@kmontemayor2-sc
Copy link
Collaborator Author

/integration_test

@kmontemayor2-sc
Copy link
Collaborator Author

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Feb 5, 2026

GiGL Automation

@ 24:33:55UTC : 🔄 Integration Test started.

@ 01:56:33UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Contributor

github-actions bot commented Feb 5, 2026

GiGL Automation

@ 24:33:57UTC : 🔄 Python Unit Test started.

@ 01:52:54UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Contributor

github-actions bot commented Feb 5, 2026

GiGL Automation

@ 24:34:02UTC : 🔄 E2E Test started.

@ 02:06:03UTC : ✅ Workflow completed successfully.

@kmontemayor2-sc
Copy link
Collaborator Author

/unit_test_py

@kmontemayor2-sc
Copy link
Collaborator Author

/integration_test

@github-actions
Copy link
Contributor

github-actions bot commented Feb 5, 2026

GiGL Automation

@ 17:43:26UTC : 🔄 Integration Test started.

@ 19:02:17UTC : ✅ Workflow completed successfully.

@github-actions
Copy link
Contributor

github-actions bot commented Feb 5, 2026

GiGL Automation

@ 17:43:26UTC : 🔄 Python Unit Test started.

@ 18:56:46UTC : ✅ Workflow completed successfully.

@kmontemayor2-sc
Copy link
Collaborator Author

/e2e_test

@github-actions
Copy link
Contributor

github-actions bot commented Feb 5, 2026

GiGL Automation

@ 17:45:54UTC : 🔄 E2E Test started.

@ 19:01:14UTC : ✅ Workflow completed successfully.

Copy link
Collaborator

@mkolodner-sc mkolodner-sc left a comment

Choose a reason for hiding this comment

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

Thanks Kyle! Left a few comments.

num_cpu_threads=num_cpu_threads,
)

self._is_input_heterogeneous = dataset_metadata.is_labeled_heterogeneous
Copy link
Collaborator

Choose a reason for hiding this comment

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

Aren't these two different things? One is asking if the input is heterogeneous, the other is checking if we are in a "labeled heterogeneous" setting.

Comment on lines -269 to +389
self._is_input_heterogeneous: bool = False

is_labeled_heterogeneous: bool = False
Copy link
Collaborator

Choose a reason for hiding this comment

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

Ditto here as well on equating these two

edge_dir=dataset.edge_dir,
seed=None, # it's actually optional - None means random.
return (
[sampler_input],
Copy link
Collaborator

Choose a reason for hiding this comment

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

Sorry I forgot why does this need to be a list again versus just the ABLPSamplerInput?

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.

3 participants