Skip to content

Fix DirectRegressor compilation + other internal changes#166

Merged
vsimkus merged 2 commits into
mainfrom
port-from-internal-052026
Jun 9, 2026
Merged

Fix DirectRegressor compilation + other internal changes#166
vsimkus merged 2 commits into
mainfrom
port-from-internal-052026

Conversation

@vsimkus

@vsimkus vsimkus commented Jun 5, 2026

Copy link
Copy Markdown
Contributor

Key changes:

  • predict() output cleanup + OrbCalculator and OrbTorchSimModel
  • forward() output cleanup + preparation for Nvalchemi OrbWrapper
  • Fixed DirectRegressor.predict() missing the compiled path. (Changed call from self.forward(batch) to self(batch).
  • Ported missing test_coulomb_module.py
  • Other misc changes

vsimkus commented Jun 5, 2026

Copy link
Copy Markdown
Contributor Author

This stack of pull requests is managed by Graphite. Learn more about stacking.

@vsimkus vsimkus marked this pull request as ready for review June 6, 2026 16:29
@vsimkus vsimkus requested a review from timduignan June 6, 2026 16:30
Comment thread orb_models/common/models/graph_regressor.py

@benrhodes26 benrhodes26 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Thanks!

@vsimkus vsimkus changed the title Port changes from internal repo Fix DirectRegressor compilation + other internal changes Jun 9, 2026
@vsimkus vsimkus merged commit 81b26ac into main Jun 9, 2026
9 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

Development

Successfully merging this pull request may close these issues.

2 participants