Skip to content

Commit 90f960c

Browse files
committed
Update doc for lammps.
1 parent a5d5afc commit 90f960c

4 files changed

Lines changed: 560 additions & 553 deletions

File tree

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ module = ["requests.*", "tabulate.*"]
190190
ignore_missing_imports = true
191191

192192
[[tool.mypy.overrides]]
193-
module = ["matgl.kernels.*", "matgl.ext.jax.*", "matgl.ext._lammps"]
193+
module = ["matgl.kernels.*", "matgl.ext.jax.*", "matgl.ext.lammps"]
194194
ignore_errors = true
195195
# ``register_buffer`` returns ``Tensor | Module`` in mypy's view; indexing,
196196
# iterating, or attribute-accessing those triggers spurious errors.

src/matgl/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ def create_lammps_model(args: argparse.Namespace) -> int:
215215
"""
216216
# Lazy import keeps the CLI responsive when this subcommand isn't used and
217217
# avoids dragging the export-only deps onto the import path.
218-
from matgl.ext._lammps import LAMMPSMatGLModel
218+
from matgl.ext.lammps import LAMMPSMatGLModel
219219

220220
dtype_map = {"float32": torch.float32, "float64": torch.float64}
221221
dtype = dtype_map[args.dtype]

0 commit comments

Comments
 (0)