-
Notifications
You must be signed in to change notification settings - Fork 3
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
ZnTrack Nodes #254
Conversation
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
apax/nodes/md.py
Outdated
# 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") |
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.
should this be removed?
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.
still testing, but probably yes. Id vote to only support
Apax(data: ase.Atoms)and remove support for
Apax(data: zntrack.Node)` - what do you think?
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.
After that it would only be possible to use Apax(AddData.atoms) but not Apax(AddData), right?
Seems fine
Move and adapt the ZnTrack Nodes from IPSuite directly into the apax package.
Missing
BAL
which requires the selection base class from IPSuite.