Skip to content

Fix deterministic seed handling, add graph utils tests, improve snaps hot prefetch concurrency, and document notebook dependencies#229

Merged
gelluisaac merged 1 commit into
Traqora:mainfrom
ritik4ever:issue-159-160-184-155
Jun 1, 2026
Merged

Fix deterministic seed handling, add graph utils tests, improve snaps hot prefetch concurrency, and document notebook dependencies#229
gelluisaac merged 1 commit into
Traqora:mainfrom
ritik4ever:issue-159-160-184-155

Conversation

@ritik4ever
Copy link
Copy Markdown
Contributor

PR Description

This PR adds deterministic training seed handling, improves graph utility coverage, enhances snapshot concurrency, and documents notebook dependency setup.

What changed

  • Added support for --seed in train.py and ASTROML_SEED fallback
  • Ensured deterministic seeding across Python, NumPy, and PyTorch
  • Added coverage for astroml.graph_utils.graph_to_pyg_data edge cases
  • Improved astroml/features/graph/snapshot.py to support parallel snapshot prefetching
  • Added notebook dependency setup instructions to example notebooks and README.md

Closes

…hot prefetch concurrency, and document notebook dependencies
@drips-wave
Copy link
Copy Markdown

drips-wave Bot commented Jun 1, 2026

@ritik4ever Great news! 🎉 Based on an automated assessment of this PR, the linked Wave issue(s) no longer count against your application limits.

You can now already apply to more issues while waiting for a review of this PR. Keep up the great work! 🚀

Learn more about application limits

@gelluisaac gelluisaac merged commit 6f51b8e into Traqora:main Jun 1, 2026
5 of 12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

2 participants