deepmd_gnn.argcheck
===================

.. py:module:: deepmd_gnn.argcheck

.. autoapi-nested-parse::

   Argument check for the MACE model.



Functions
---------

.. autoapisummary::

   deepmd_gnn.argcheck.mace_model_args
   deepmd_gnn.argcheck.nequip_model_args


Module Contents
---------------

.. py:function:: mace_model_args() -> dargs.Argument

   Arguments for the MACE model.

   Returns
   -------
   Argument
       Arguments for the MACE model.


.. py:function:: nequip_model_args() -> dargs.Argument

   Arguments for the NequIP model.