Skip to content

Fix a small bug in NativePET preventing training with arbitrary targets #546

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Apr 14, 2025

Conversation

frostedoyster
Copy link
Collaborator

@frostedoyster frostedoyster commented Apr 9, 2025


📚 Documentation preview 📚: https://metatrain--546.org.readthedocs.build/en/546/

Copy link

@Copilot Copilot AI left a comment

Choose a reason for hiding this comment

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

Copilot reviewed 2 out of 2 changed files in this pull request and generated 1 comment.

Comments suppressed due to low confidence (1)

src/metatrain/experimental/nativepet/model.py:538

  • Direct assignment for edge predictions with the same key risks overwriting data when multiple predictions exist. Accumulating these predictions in a list might be more appropriate.
edge_atomic_predictions_dict[output_name] = (

@frostedoyster frostedoyster force-pushed the fix-nativepet-arbitrary-targets branch from db29379 to 5f9d18c Compare April 14, 2025 13:30
@frostedoyster frostedoyster force-pushed the fix-nativepet-arbitrary-targets branch from 5f9d18c to 7901b6f Compare April 14, 2025 13:47
@frostedoyster frostedoyster merged commit 6b8298c into main Apr 14, 2025
11 of 12 checks passed
@frostedoyster frostedoyster deleted the fix-nativepet-arbitrary-targets branch April 14, 2025 13:47
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