NequIPFrameworkModel

class torch_sim.models.nequip_framework.NequIPFrameworkModel(err=_nequip_import_error, *_args, **_kwargs)

Bases: ModelInterface

NequIP model framework wrapper for torch-sim.

NOTE:This class is a placeholder when NequIP is not installed. It raises an ImportError if accessed.

Parameters:

• err (ImportError)

• _args (Any)

• _kwargs (Any)

classmethod from_compiled_model(_path, *_args, **_kwargs)

Dummy classmethod for type checking when NequIP is not installed.

Parameters:

• _path (Any)

• _args (Any)

• _kwargs (Any)

Return type:

Self