-
Notifications
You must be signed in to change notification settings - Fork 19
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
rework handle_trajectories to use replicas #539
Conversation
functions now expect a replica id, can trivially convert from state id to replica id trajectory_from_multistate can now either do constant-state or constant-replica trajectories
Codecov ReportPatch coverage:
Additional details and impacted files@@ Coverage Diff @@
## main #539 +/- ##
==========================================
- Coverage 92.07% 90.98% -1.09%
==========================================
Files 113 114 +1
Lines 6938 6876 -62
==========================================
- Hits 6388 6256 -132
- Misses 550 620 +70
☔ View full report in Codecov by Sentry. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just two small things, otherwise lgtm!
openfe/utils/handle_trajectories.py
Outdated
@@ -42,7 +42,7 @@ def _state_positions_at_frame(dataset: nc.Dataset, state_num: int, | |||
---------- | |||
dataset : netCDF4.Dataset | |||
Dataset containing the MultiState information. | |||
state_num : int | |||
replica_id : int | |||
State index to extract positions for. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Shoudl this be changed to replica index?
openfe/utils/handle_trajectories.py
Outdated
@@ -33,7 +32,8 @@ def _effective_replica(dataset: nc.Dataset, state_num: int, | |||
return np.where(state_distribution == state_num)[0][0] | |||
|
|||
|
|||
def _state_positions_at_frame(dataset: nc.Dataset, state_num: int, | |||
def _state_positions_at_frame(dataset: nc.Dataset, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
replica_positions_at_frame?
functions now expect a replica id, can trivially convert from state id to replica id
trajectory_from_multistate can now either do constant-state or constant-replica trajectories
Developers certificate of origin