bump monty to use the monty.json import speedup patch, add import time regression test, lazy load some rarely used but costly modules

pyproject.toml

@@ -56,8 +56,8 @@ classifiers = [
 dependencies = [
     "joblib>=1",
     "matplotlib>=3.8",
-    "monty>=2024.7.29",
-    "networkx>=3",  # PR #4116
+    "monty>=2024.10.21",
+    "networkx>=2.7",  # PR #4116
     "palettable>=3.3.3",
     "pandas>=2",
     "plotly>=4.5.0",
@@ -69,7 +69,7 @@ dependencies = [
     # https://github.com/scipy/scipy/issues/21052
     "scipy>=1.14.1; platform_system == 'Windows'",
     "spglib>=2.5.0",
-    "sympy>=1.2",
+    "sympy>=1.3",  # PR #4116
     "tabulate>=0.9",
     "tqdm>=4.60",
     "uncertainties>=3.1.4",
@@ -87,40 +87,34 @@ Issues = "https://github.com/materialsproject/pymatgen/issues"
 Pypi = "https://pypi.org/project/pymatgen"
 
 [project.optional-dependencies]
+# PR4128: netcdf4 1.7.[0/1] yanked, 1.7.1.post[1/2]/1.7.2 cause CI error
+abinit = ["netcdf4>=1.6.5,!=1.7.1.post1,!=1.7.1.post2,!=1.7.2"]
 ase = ["ase>=3.23.0"]
-# tblite only support Python 3.12+ through conda-forge
-# https://github.com/tblite/tblite/issues/175
-tblite = ["tblite[ase]>=0.3.0; python_version<'3.12'"]
-vis = ["vtk>=6.0.0"]
-abinit = ["netcdf4>=1.7.1"]
-mlp = ["chgnet>=0.3.8", "matgl>=1.1.3"]
-electronic_structure = ["fdint>=2.0.2"]
 ci = ["pytest-cov>=4", "pytest-split>=0.8", "pytest>=8"]
 docs = ["invoke", "sphinx", "sphinx_markdown_builder", "sphinx_rtd_theme"]
+electronic_structure = ["fdint>=2.0.2"]
+mlp = ["chgnet>=0.3.8", "matgl>=1.1.3"]
+numba = ["numba>=0.55"]
+numpy-v1 = ["numpy>=1.25.0,<2"] # Test NP1 on Windows (quite buggy ATM)
 optional = [
-    "ase>=3.23.0",
+    "pymatgen[abinit,ase,mlp,tblite]",
     "beautifulsoup4",
     # BoltzTraP2 build fails on Windows GitHub runners
     "BoltzTraP2>=24.9.4 ; platform_system != 'Windows'",
     "chemview>=0.6",
-    "chgnet>=0.3.8",
     "f90nml>=1.1.2",
     "galore>=0.6.1",
     "h5py>=3.11.0",
+    "hiphive>=1.3.1",
     "jarvis-tools>=2020.7.14",
-    "matgl>=1.1.3",
     "matplotlib>=3.8",
-    "netCDF4>=1.6.5",
     "phonopy>=2.23",
     "seekpath>=2.0.1",
-    # tblite only support Python 3.12+ through conda-forge
-    # https://github.com/tblite/tblite/issues/175
-    "hiphive>=1.3.1",
-    "openbabel-wheel>=3.1.1.20",
-    "tblite[ase]>=0.3.0; platform_system=='Linux' and python_version<'3.12'",
 ]
-numba = ["numba>=0.55"]
-numpy-v1 = ["numpy>=1.25.0,<2"] # Test NP1 on Windows (quite buggy ATM)
+# tblite only support Python 3.12+ through conda-forge
+# https://github.com/tblite/tblite/issues/175
+tblite = [ "tblite[ase]>=0.3.0; platform_system=='Linux' and python_version<'3.12'"]
+vis = ["vtk>=6.0.0"]
 
 [project.scripts]
 pmg = "pymatgen.cli.pmg:main"

src/pymatgen/analysis/adsorption.py

@@ -9,8 +9,6 @@
 from typing import TYPE_CHECKING
 
 import numpy as np
-from matplotlib import patches
-from matplotlib.path import Path
 from monty.serialization import loadfn
 from scipy.spatial import Delaunay
 
@@ -664,6 +662,10 @@ def plot_slab(
         decay (float): how the alpha-value decays along the z-axis
         inverse (bool): invert z axis to plot opposite surface
     """
+    # Expensive import (PR4128)
+    from matplotlib import patches
+    from matplotlib.path import Path
+
     orig_slab = slab.copy()
     slab = reorient_z(slab)
     orig_cell = slab.lattice.matrix.copy()

src/pymatgen/analysis/interfaces/coherent_interfaces.py

@@ -6,7 +6,6 @@
 from typing import TYPE_CHECKING
 
 import numpy as np
-from numpy.testing import assert_allclose
 from scipy.linalg import polar
 
 from pymatgen.analysis.elasticity.strain import Deformation
@@ -101,21 +100,15 @@ def _find_matches(self) -> None:
         for match in self.zsl_matches:
             xform = get_2d_transform(film_vectors, match.film_vectors)
             strain, _rot = polar(xform)
-            assert_allclose(
-                strain,
-                np.round(strain),
-                atol=1e-12,
-                err_msg="Film lattice vectors changed during ZSL match, check your ZSL Generator parameters",
-            )
+            if not np.allclose(strain, np.round(strain), atol=1e-12):
+                raise ValueError("Film lattice vectors changed during ZSL match, check your ZSL Generator parameters")
 
             xform = get_2d_transform(substrate_vectors, match.substrate_vectors)
             strain, _rot = polar(xform)
-            assert_allclose(
-                strain,
-                strain.astype(int),
-                atol=1e-12,
-                err_msg="Substrate lattice vectors changed during ZSL match, check your ZSL Generator parameters",
-            )
+            if not np.allclose(strain, strain.astype(int), atol=1e-12):
+                raise ValueError(
+                    "Substrate lattice vectors changed during ZSL match, check your ZSL Generator parameters"
+                )
 
     def _find_terminations(self):
         """Find all terminations."""
@@ -226,37 +219,23 @@ def get_interfaces(
             ).astype(int)
             film_sl_slab = film_slab.copy()
             film_sl_slab.make_supercell(super_film_transform)
-            assert_allclose(
-                film_sl_slab.lattice.matrix[2],
-                film_slab.lattice.matrix[2],
-                atol=1e-08,
-                err_msg="2D transformation affected C-axis for Film transformation",
-            )
-            assert_allclose(
-                film_sl_slab.lattice.matrix[:2],
-                match.film_sl_vectors,
-                atol=1e-08,
-                err_msg="Transformation didn't make proper supercell for film",
-            )
+            if not np.allclose(film_sl_slab.lattice.matrix[2], film_slab.lattice.matrix[2], atol=1e-08):
+                raise ValueError(
+                    "2D transformation affected C-axis for Film transformation",
+                )
+            if not np.allclose(film_sl_slab.lattice.matrix[:2], match.film_sl_vectors, atol=1e-08):
+                raise ValueError("Transformation didn't make proper supercell for film")
 
             # Build substrate superlattice
             super_sub_transform = np.round(
                 from_2d_to_3d(get_2d_transform(sub_slab.lattice.matrix[:2], match.substrate_sl_vectors))
             ).astype(int)
             sub_sl_slab = sub_slab.copy()
             sub_sl_slab.make_supercell(super_sub_transform)
-            assert_allclose(
-                sub_sl_slab.lattice.matrix[2],
-                sub_slab.lattice.matrix[2],
-                atol=1e-08,
-                err_msg="2D transformation affected C-axis for Film transformation",
-            )
-            assert_allclose(
-                sub_sl_slab.lattice.matrix[:2],
-                match.substrate_sl_vectors,
-                atol=1e-08,
-                err_msg="Transformation didn't make proper supercell for substrate",
-            )
+            if not np.allclose(sub_sl_slab.lattice.matrix[2], sub_slab.lattice.matrix[2], atol=1e-08):
+                raise ValueError("2D transformation affected C-axis for Film transformation")
+            if not np.allclose(sub_sl_slab.lattice.matrix[:2], match.substrate_sl_vectors, atol=1e-08):
+                raise ValueError("Transformation didn't make proper supercell for substrate")
 
             # Add extra info
             match_dict = match.as_dict()

src/pymatgen/core/__init__.py

@@ -5,7 +5,7 @@
 import os
 import warnings
 from importlib.metadata import PackageNotFoundError, version
-from typing import Any
+from typing import TYPE_CHECKING
 
 from ruamel.yaml import YAML
 
@@ -17,6 +17,9 @@
 from pymatgen.core.structure import IMolecule, IStructure, Molecule, PeriodicNeighbor, SiteCollection, Structure
 from pymatgen.core.units import ArrayWithUnit, FloatWithUnit, Unit
 
+if TYPE_CHECKING:
+    from typing import Any
+
 __author__ = "Pymatgen Development Team"
 __email__ = "[email protected]"
 __maintainer__ = "Shyue Ping Ong, Matthew Horton, Janosh Riebesell"

src/pymatgen/core/interface.py

@@ -14,7 +14,6 @@
 
 import numpy as np
 from monty.fractions import lcm
-from numpy.testing import assert_allclose
 from scipy.cluster.hierarchy import fcluster, linkage
 from scipy.spatial.distance import squareform
 
@@ -2784,10 +2783,16 @@ def from_slabs(
             substrate_slab = substrate_slab.get_orthogonal_c_slab()
         if isinstance(film_slab, Slab):
             film_slab = film_slab.get_orthogonal_c_slab()
-        assert_allclose(film_slab.lattice.alpha, 90, 0.1)
-        assert_allclose(film_slab.lattice.beta, 90, 0.1)
-        assert_allclose(substrate_slab.lattice.alpha, 90, 0.1)
-        assert_allclose(substrate_slab.lattice.beta, 90, 0.1)
+
+        if not math.isclose(film_slab.lattice.alpha, 90, rel_tol=0.1) or not math.isclose(
+            film_slab.lattice.beta, 90, rel_tol=0.1
+        ):
+            raise ValueError("The lattice angles alpha or beta of the film slab are not approximately 90 degrees.")
+
+        if not math.isclose(substrate_slab.lattice.alpha, 90, rel_tol=0.1) or not math.isclose(
+            substrate_slab.lattice.beta, 90, rel_tol=0.1
+        ):
+            raise ValueError("The lattice angles alpha or beta of the substrate slab are not approximately 90 degrees.")
 
         # Ensure sub is right-handed
         # IE sub has surface facing "up"

src/pymatgen/core/operations.py

@@ -449,7 +449,7 @@ def as_xyz_str(self) -> str:
         Only works for integer rotation matrices.
         """
         # Check for invalid rotation matrix
-        if not np.all(np.isclose(self.rotation_matrix, np.round(self.rotation_matrix))):
+        if not np.allclose(self.rotation_matrix, np.round(self.rotation_matrix)):
             warnings.warn("Rotation matrix should be integer")
 
         return transformation_to_string(

src/pymatgen/core/trajectory.py

@@ -15,7 +15,6 @@
 from monty.json import MSONable
 
 from pymatgen.core.structure import Composition, DummySpecies, Element, Lattice, Molecule, Species, Structure
-from pymatgen.io.ase import AseAtomsAdaptor
 
 if TYPE_CHECKING:
     from collections.abc import Iterator
@@ -580,9 +579,12 @@ def from_file(cls, filename: str | Path, constant_lattice: bool = True, **kwargs
             try:
                 from ase.io.trajectory import Trajectory as AseTrajectory
 
+                from pymatgen.io.ase import AseAtomsAdaptor
+
                 ase_traj = AseTrajectory(filename)
                 # Periodic boundary conditions should be the same for all frames so just check the first
                 pbc = ase_traj[0].pbc
+
                 if any(pbc):
                     structures = [AseAtomsAdaptor.get_structure(atoms) for atoms in ase_traj]
                 else:

src/pymatgen/io/vasp/outputs.py

@@ -24,7 +24,6 @@
 from monty.json import MSONable, jsanitize
 from monty.os.path import zpath
 from monty.re import regrep
-from numpy.testing import assert_allclose
 from tqdm import tqdm
 
 from pymatgen.core import Composition, Element, Lattice, Structure
@@ -4978,8 +4977,8 @@ def __init__(
 
                     if i_spin == 0:
                         self.kpoints.append(kpoint)
-                    else:
-                        assert_allclose(self.kpoints[i_nk], kpoint)
+                    elif not np.allclose(self.kpoints[i_nk], kpoint, rtol=1e-7, atol=0):
+                        raise ValueError(f"kpoints of {i_nk=} mismatch")
 
                     if verbose:
                         print(f"kpoint {i_nk: 4} with {nplane: 5} plane waves at {kpoint}")

src/pymatgen/symmetry/settings.py

@@ -7,8 +7,6 @@
 from typing import TYPE_CHECKING
 
 import numpy as np
-from sympy import Matrix
-from sympy.parsing.sympy_parser import parse_expr
 
 from pymatgen.core.lattice import Lattice
 from pymatgen.core.operations import MagSymmOp, SymmOp
@@ -100,6 +98,10 @@ def parse_transformation_string(
         Returns:
             tuple[list[list[float]] | np.ndarray, list[float]]: transformation matrix & vector
         """
+        # Import sympy is expensive (PR4128)
+        from sympy import Matrix
+        from sympy.parsing.sympy_parser import parse_expr
+
         try:
             a, b, c = np.eye(3)
             b_change, o_shift = transformation_string.split(";")

tests/io/abinit/test_netcdf.py

@@ -92,6 +92,8 @@ def test_read_fe(self):
                 ref_magmom_collinear = [-0.5069359730980665]
                 path = os.path.join(tmp_dir, "Fe_magmoms_collinear_GSR.nc")
 
+                # TODO: PR4128, EtsfReader would fail in Ubuntu CI with netCDF4 > 1.6.5
+                # Need someone with knowledge in netCDF4 to fix it
                 with EtsfReader(path) as data:
                     structure = data.read_structure()
                     assert structure.site_properties["magmom"] == ref_magmom_collinear

tests/performance/test_performance.py

@@ -0,0 +1,14 @@
+"""
+Test the import time of several important modules.
+
+TODO:
+    Test the runtime of several important modules.
+"""
+
+from __future__ import annotations
+
+
+class TestImportTime:
+    """
+    Test import time of important modules.
+    """