Merge remote-tracking branch 'upstream/master'

Author: esoteric-ephemera

.coderabbit.yaml

@@ -111,11 +111,11 @@ def main():
     del lines[start + 1 : stop]
 
     # [2] write new py module
-    with open(xc_funcpy_path, mode="w") as file:
+    with open(xc_funcpy_path, mode="w", encoding="utf-8") as file:
         file.writelines(lines)
 
     print("Files have been regenerated")
-    print("Remember to update libxc_version in libxcfuncs.py!")
+    print("Remember to update __version__ in libxcfuncs.py!")
 
     return 0
 

dev_scripts/regen_libxcfunc.py

@@ -914,8 +914,7 @@ def find_eq_stress(strains, stresses, tol: float = 1e-10):
 
 
 def get_strain_state_dict(strains, stresses, eq_stress=None, tol: float = 1e-10, add_eq=True, sort=True):
-    """
-    Creates a dictionary of voigt notation stress-strain sets
+    """Create a dictionary of voigt notation stress-strain sets
     keyed by "strain state", i. e. a tuple corresponding to
     the non-zero entries in ratios to the lowest nonzero value,
     e.g. [0, 0.1, 0, 0.2, 0, 0] -> (0,1,0,2,0,0)

pymatgen/analysis/elasticity/elastic.py

@@ -541,17 +541,15 @@ def __init__(self, matrix, m_list, num_to_return=1, algo=ALGO_FAST):
         self._minimized_sum = self._output_lists[0][0]
 
     def minimize_matrix(self):
-        """
-        This method finds and returns the permutations that produce the lowest
+        """Get the permutations that produce the lowest
         Ewald sum calls recursive function to iterate through permutations.
         """
         if self._algo in (EwaldMinimizer.ALGO_FAST, EwaldMinimizer.ALGO_BEST_FIRST):
             return self._recurse(self._matrix, self._m_list, set(range(len(self._matrix))))
         return None
 
     def add_m_list(self, matrix_sum, m_list):
-        """
-        This adds an m_list to the output_lists and updates the current
+        """Add an m_list to the output_lists and updates the current
         minimum if the list is full.
         """
         if self._output_lists is None:
@@ -629,9 +627,7 @@ def get_next_index(cls, matrix, manipulation, indices_left):
         return indices[sums.argmax(axis=0)] if f < 1 else indices[sums.argmin(axis=0)]
 
     def _recurse(self, matrix, m_list, indices, output_m_list=None):
-        """
-        This method recursively finds the minimal permutations using a binary
-        tree search strategy.
+        """Find the minimal permutations using a binary tree search strategy.
 
         Args:
             matrix: The current matrix (with some permutations already

pymatgen/analysis/ewald.py

@@ -70,8 +70,7 @@
 
 
 def zval_dict_from_potcar(potcar) -> dict[str, float]:
-    """
-    Creates zval_dictionary for calculating the ionic polarization from
+    """Create zval_dictionary for calculating the ionic polarization from
     Potcar object.
 
     potcar: Potcar object

pymatgen/analysis/ferroelectricity/polarization.py

@@ -604,9 +604,7 @@ def __init__(
         truncate_by_symmetry: bool = True,
         transformation_kwargs: dict | None = None,
     ):
-        """
-        This class will try generated different collinear
-        magnetic orderings for a given input structure.
+        """Generate different collinear magnetic orderings for a given input structure.
 
         If the input structure has magnetic moments defined, it
         is possible to use these as a hint as to which elements are

pymatgen/analysis/magnetism/analyzer.py

@@ -669,10 +669,8 @@ class HeisenbergScreener:
     """Clean and screen magnetic orderings."""
 
     def __init__(self, structures, energies, screen=False):
-        """
-        This class pre-processes magnetic orderings and energies for
-        HeisenbergMapper. It prioritizes low-energy orderings with large and
-        localized magnetic moments.
+        """Pre-processes magnetic orderings and energies for HeisenbergMapper.
+        It prioritizes low-energy orderings with large and localized magnetic moments.
 
         Args:
             structures (list): Structure objects with magnetic moments.

pymatgen/analysis/magnetism/heisenberg.py

@@ -45,9 +45,7 @@ class Substitutor(MSONable):
     charge_balanced_tol: float = 1e-9
 
     def __init__(self, threshold=1e-3, symprec: float = 0.1, **kwargs):
-        """
-        This substitutor uses the substitution probability class to
-        find good substitutions for a given chemistry or structure.
+        """Use the substitution probability class to find good substitutions for a given chemistry or structure.
 
         Args:
             threshold:

pymatgen/analysis/structure_prediction/substitutor.py

@@ -671,8 +671,7 @@ def anisotropy(self) -> float:
 
     @property
     def shape_factor(self) -> float:
-        """
-        This is useful for determining the critical nucleus size.
+        """Determine the critical nucleus size.
         A large shape factor indicates great anisotropy.
         See Ballufi, R. W., Allen, S. M. & Carter, W. C. Kinetics
             of Materials. (John Wiley & Sons, 2005), p.461.
@@ -685,10 +684,10 @@ def shape_factor(self) -> float:
     @property
     def effective_radius(self) -> float:
         """
-        Radius of the WulffShape when the WulffShape is approximated as a sphere.
+        Radius of the WulffShape (in Angstroms) when the WulffShape is approximated as a sphere.
 
         Returns:
-            float: radius.
+            float: radius R_eff
         """
         return ((3 / 4) * (self.volume / np.pi)) ** (1 / 3)
 

pymatgen/analysis/wulff.py

@@ -1,4 +1,4 @@
-"""This class implements definitions for various kinds of bonds. Typically used in
+"""This module implements definitions for various kinds of bonds. Typically used in
 Molecule analysis.
 """
 
@@ -17,10 +17,13 @@
     from pymatgen.util.typing import SpeciesLike
 
 
-def _load_bond_length_data():
-    """Loads bond length data from json file."""
-    with open(os.path.join(os.path.dirname(__file__), "bond_lengths.json")) as file:
-        data = defaultdict(dict)
+def _load_bond_length_data() -> dict[tuple[str, ...], dict[float, float]]:
+    """Load bond length data from bond_lengths.json file."""
+    with open(
+        os.path.join(os.path.dirname(__file__), "bond_lengths.json"),
+        encoding="utf-8",
+    ) as file:
+        data: dict[tuple, dict] = defaultdict(dict)
         for row in json.load(file):
             els = sorted(row["elements"])
             data[tuple(els)][row["bond_order"]] = row["length"]
@@ -43,12 +46,19 @@ def __init__(self, site1: Site, site2: Site) -> None:
         self.site1 = site1
         self.site2 = site2
 
+    def __repr__(self) -> str:
+        return f"Covalent bond between {self.site1} and {self.site2}"
+
     @property
     def length(self) -> float:
         """Length of the bond."""
         return self.site1.distance(self.site2)
 
-    def get_bond_order(self, tol: float = 0.2, default_bl: float | None = None) -> float:
+    def get_bond_order(
+        self,
+        tol: float = 0.2,
+        default_bl: float | None = None,
+    ) -> float:
         """The bond order according the distance between the two sites.
 
         Args:
@@ -71,8 +81,14 @@ def get_bond_order(self, tol: float = 0.2, default_bl: float | None = None) -> f
         return get_bond_order(sp1, sp2, dist, tol, default_bl)
 
     @staticmethod
-    def is_bonded(site1, site2, tol: float = 0.2, bond_order: float | None = None, default_bl: float | None = None):
-        """Test if two sites are bonded, up to a certain limit.
+    def is_bonded(
+        site1: Site,
+        site2: Site,
+        tol: float = 0.2,
+        bond_order: float | None = None,
+        default_bl: float | None = None,
+    ) -> bool:
+        """Check if two sites are bonded, up to a certain limit.
 
         Args:
             site1 (Site): First site
@@ -87,7 +103,7 @@ def is_bonded(site1, site2, tol: float = 0.2, bond_order: float | None = None, d
                 bond length. If None, a ValueError will be thrown.
 
         Returns:
-            Boolean indicating whether two sites are bonded.
+            bool: whether two sites are bonded.
         """
         sp1 = next(iter(site1.species))
         sp2 = next(iter(site2.species))
@@ -102,11 +118,12 @@ def is_bonded(site1, site2, tol: float = 0.2, bond_order: float | None = None, d
             return dist < (1 + tol) * default_bl
         raise ValueError(f"No bond data for elements {syms[0]} - {syms[1]}")
 
-    def __repr__(self) -> str:
-        return f"Covalent bond between {self.site1} and {self.site2}"
-
 
-def obtain_all_bond_lengths(sp1, sp2, default_bl: float | None = None):
+def obtain_all_bond_lengths(
+    sp1: SpeciesLike,
+    sp2: SpeciesLike,
+    default_bl: float | None = None,
+) -> dict[float, float]:
     """Obtain bond lengths for all bond orders from bond length database.
 
     Args:
@@ -127,17 +144,23 @@ def obtain_all_bond_lengths(sp1, sp2, default_bl: float | None = None):
     if syms in bond_lengths:
         return bond_lengths[syms].copy()
     if default_bl is not None:
-        return {1: default_bl}
+        return {1.0: default_bl}
     raise ValueError(f"No bond data for elements {syms[0]} - {syms[1]}")
 
 
-def get_bond_order(sp1, sp2, dist: float, tol: float = 0.2, default_bl: float | None = None):
+def get_bond_order(
+    sp1: SpeciesLike,
+    sp2: SpeciesLike,
+    dist: float,
+    tol: float = 0.2,
+    default_bl: float | None = None,
+) -> float:
     """Calculate the bond order given the distance of 2 species.
 
     Args:
         sp1 (Species): First specie.
         sp2 (Species): Second specie.
-        dist: Their distance in angstrom
+        dist (float): Distance in angstrom
         tol (float): Relative tolerance to test. Basically, the code
             checks if the distance between the sites is larger than
             (1 + tol) * the longest bond distance or smaller than
@@ -148,8 +171,7 @@ def get_bond_order(sp1, sp2, dist: float, tol: float = 0.2, default_bl: float |
             bond length (bond order = 1). If None, a ValueError will be thrown.
 
     Returns:
-        Float value of bond order. For example, for C-C bond in benzene,
-        return 1.7.
+        float: Bond order. For example, 1.7 for C-C bond in benzene.
     """
     all_lens = obtain_all_bond_lengths(sp1, sp2, default_bl)
     # Transform bond lengths dict to list assuming bond data is successive
@@ -172,7 +194,11 @@ def get_bond_order(sp1, sp2, dist: float, tol: float = 0.2, default_bl: float |
     return trial_bond_order - 1
 
 
-def get_bond_length(sp1: SpeciesLike, sp2: SpeciesLike, bond_order: float = 1) -> float:
+def get_bond_length(
+    sp1: SpeciesLike,
+    sp2: SpeciesLike,
+    bond_order: float = 1,
+) -> float:
     """Get the bond length between two species.
 
     Args:
@@ -184,8 +210,8 @@ def get_bond_length(sp1: SpeciesLike, sp2: SpeciesLike, bond_order: float = 1) -
             C-C bond length, this should be set to 2. Defaults to 1.
 
     Returns:
-        Bond length in Angstrom. If no data is available, the sum of the atomic
-        radius is used.
+        float: Bond length in Angstrom. If no data is available,
+            the sum of the atomic radius is used.
     """
     sp1 = Element(sp1) if isinstance(sp1, str) else sp1
     sp2 = Element(sp2) if isinstance(sp2, str) else sp2