Source code for deepmd.entrypoints.doc
"""Module that prints train input arguments docstrings."""
from deepmd.utils.argcheck import gen_doc, gen_json
__all__ = ["doc_train_input"]
[docs]def doc_train_input(*, out_type: str = "rst", **kwargs):
"""Print out trining input arguments to console."""
if out_type == "rst":
doc_str = gen_doc(make_anchor=True)
elif out_type == "json":
doc_str = gen_json()
else:
raise RuntimeError("Unsupported out type %s" % out_type)
print(doc_str)