deepmd_gnn.mace_off_cli
=======================

.. py:module:: deepmd_gnn.mace_off_cli

.. autoapi-nested-parse::

   Lightweight helpers for official MACE-OFF checkpoint selection and download.



Attributes
----------

.. autoapisummary::

   deepmd_gnn.mace_off_cli.MACE_OFF_MODELS
   deepmd_gnn.mace_off_cli.MACE_OFF_MODEL_URLS
   deepmd_gnn.mace_off_cli.MACE_OFF_MODEL_SHA256
   deepmd_gnn.mace_off_cli.MACE_OFF_MODEL_CHOICES


Functions
---------

.. autoapisummary::

   deepmd_gnn.mace_off_cli.get_mace_off_cache_dir
   deepmd_gnn.mace_off_cli.download_mace_off_model


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

.. py:data:: MACE_OFF_MODELS

.. py:data:: MACE_OFF_MODEL_URLS

.. py:data:: MACE_OFF_MODEL_SHA256

.. py:data:: MACE_OFF_MODEL_CHOICES

.. py:function:: get_mace_off_cache_dir() -> pathlib.Path

   Get the cache directory for MACE-OFF models.


.. py:function:: download_mace_off_model(model_name: str, cache_dir: pathlib.Path | None = None) -> pathlib.Path

   Download a selected official MACE-OFF model file.

   Parameters
   ----------
   model_name
       Canonical model name (for example ``off23_small``) or one of the
       compatibility aliases ``small`` / ``medium`` / ``large``.
   cache_dir
       Cache directory. Defaults to :func:`get_mace_off_cache_dir`.