Skip to content

Commit

Permalink
Add somd2 and sire versions to parquet file metadata.
Browse files Browse the repository at this point in the history
  • Loading branch information
lohedges committed Apr 8, 2024
1 parent af5dc23 commit 9954b82
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions src/somd2/runner/_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -938,12 +938,16 @@ def _run(sim, is_restart=False):
raise
self._sim._cleanup()

from somd2 import __version__, _sire_version, _sire_revisionid

# Write final dataframe for the system to the energy trajectory file.
# Note that sire s3 checkpoint files contain energy trajectory data, so this works even for restarts.
_ = _dataframe_to_parquet(
df,
metadata={
"attrs": df.attrs,
"somd2 version": __version__,
"sire version": f"{_sire_version}+{_sire_revisionid}",
"lambda": str(lambda_value),
"lambda_array": self._lambda_values,
"lambda_grad": lambda_grad,
Expand Down

0 comments on commit 9954b82

Please sign in to comment.