Skip to content
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

ZnTrack Nodes #254

Merged
merged 25 commits into from
Apr 5, 2024
Merged

ZnTrack Nodes #254

merged 25 commits into from
Apr 5, 2024

Conversation

PythonFZ
Copy link
Contributor

@PythonFZ PythonFZ commented Apr 3, 2024

Move and adapt the ZnTrack Nodes from IPSuite directly into the apax package.

  • tests
  • adapt and cleanup

Missing BAL which requires the selection base class from IPSuite.

apax/nodes/md.py Outdated
Comment on lines 43 to 65
# def post_init(self):
# if not self.state.loaded:
# dict_supplied = self.md_parameter is None
# file_supplied = self.md_parameter_file is None

# if dict_supplied and file_supplied:
# raise TypeError("Please specify either an input dict or a file")
# elif not dict_supplied and not file_supplied:
# raise TypeError(
# "Can not train apax model without a parameter dict or file"
# )
# else:
# log.info(
# "Please keep track of the parameter file with git, just like the"
# " params.yaml"
# )

# if not isinstance(self.model, Apax):
# raise TypeError(
# "Performing simulations with JaxMD requires a apax model Node"
# )

# self.data = utils.helpers.get_deps_if_node(self.data, "atoms")
Copy link
Contributor

Choose a reason for hiding this comment

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

should this be removed?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

still testing, but probably yes. Id vote to only support Apax(data: ase.Atoms)and remove support forApax(data: zntrack.Node)` - what do you think?

Copy link
Contributor

Choose a reason for hiding this comment

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

After that it would only be possible to use Apax(AddData.atoms) but not Apax(AddData), right?
Seems fine

apax/nodes/model.py Outdated Show resolved Hide resolved
@PythonFZ PythonFZ requested a review from M-R-Schaefer April 5, 2024 09:42
@PythonFZ PythonFZ marked this pull request as ready for review April 5, 2024 09:53
@M-R-Schaefer M-R-Schaefer merged commit 6ce9021 into main Apr 5, 2024
3 checks passed
@M-R-Schaefer M-R-Schaefer deleted the zntrack branch April 5, 2024 11:15
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