Skip to content

Split constraints#547

Draft
thomasloux wants to merge 1 commit intoTorchSim:mainfrom
thomasloux:feat/add-split-constraints
Draft

Split constraints#547
thomasloux wants to merge 1 commit intoTorchSim:mainfrom
thomasloux:feat/add-split-constraints

Conversation

@thomasloux
Copy link
Copy Markdown
Collaborator

Summary

More efficient, especially for FixSymmetry which has many variables, way to split states.

Checklist

Before a pull request can be merged, the following items must be checked:

  • Doc strings have been added in the Google docstring format.
  • Run ruff on your code.
  • Tests have been added for any new functionality or bug fixes.

@thomasloux thomasloux marked this pull request as draft April 16, 2026 16:10
@thomasloux
Copy link
Copy Markdown
Collaborator Author

I need to review and test.

Comment thread torch_sim/constraints.py
result[sys_idx] = type(self)(
[self.rotations[local_idx]],
[self.symm_maps[local_idx]],
zero.clone(),
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

can we use kwargs here? it's hard to understand what you are setting to zero.

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