phonopy 3.2.0__tar.gz → 3.2.1__tar.gz

{phonopy-3.2.0 → phonopy-3.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: phonopy
-Version: 3.2.0
+Version: 3.2.1
 Summary: This is the phonopy module.
 Author-Email: Atsushi Togo <atz.togo@gmail.com>
 License-Expression: BSD-3-Clause

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '3.2.0'
-__version_tuple__ = version_tuple = (3, 2, 0)
+__version__ = version = '3.2.1'
+__version_tuple__ = version_tuple = (3, 2, 1)
 
-__commit_id__ = commit_id = 'gd443de28b'
+__commit_id__ = commit_id = 'gbc7b2c188'

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/api_phonopy.py

@@ -256,7 +256,7 @@ class Phonopy:
         self._search_primitive_symmetry()
 
         # displacements
-        self._dataset = None
+        self._dataset: DisplacementDataset | None = None
         self._supercells_with_displacements = None
 
         # set_force_constants or set_forces
@@ -270,7 +270,7 @@ class Phonopy:
 
         # MLP
         self._mlp = None
-        self._mlp_dataset = None
+        self._mlp_dataset: Type2DisplacementDataset | None = None
 
         self._band_structure = None
         self._mesh = None
@@ -404,13 +404,13 @@ class Phonopy:
         return self._calculator
 
     @property
-    def dataset(self) -> dict | None:
+    def dataset(self) -> DisplacementDataset | None:
         """Return displacement-force dataset.
 
         Dataset containing information of displacements in supercells.
         This optionally contains energies and forces of respective supercells.
 
-        dataset : dict
+        dataset : DisplacementDataset or None
             The format can be either one of two types
 
             Type 1. One atomic displacement in each supercell:
@@ -441,13 +441,13 @@ class Phonopy:
         return self._dataset
 
     @dataset.setter
-    def dataset(self, dataset: dict | None) -> None:
+    def dataset(self, dataset: DisplacementDataset | None) -> None:
         if dataset is None:
             self._dataset = None
         elif "first_atoms" in dataset:
             self._dataset = copy.deepcopy(dataset)
         elif "displacements" in dataset:
-            self._dataset = {}
+            self._dataset = None
             self.displacements = dataset["displacements"]
             if "forces" in dataset:
                 self.forces = dataset["forces"]
@@ -552,14 +552,11 @@ class Phonopy:
         disp = np.array(displacements, dtype="double", order="C")
         if disp.ndim != 3 or disp.shape[1:] != (len(self._supercell), 3):
             raise RuntimeError("Array shape of displacements is incorrect.")
-        if self._dataset is None:
-            self._dataset = {}
-        if "first_atoms" in self._dataset:
+        if self._dataset is not None and "first_atoms" in self._dataset:
             raise RuntimeError(
                 "Setting displacements to type-1 dataset is not supported."
             )
-
-        self._dataset["displacements"] = disp
+        self._dataset = {"displacements": disp}
         self._supercells_with_displacements = None
 
     @property
@@ -3285,8 +3282,7 @@ class Phonopy:
             phpy_yaml = PhonopyYaml(
                 configuration=configuration, physical_units=units, settings=settings
             )
-        phpy_yaml.frequency_unit_conversion_factor = self._unit_conversion_factor
-        set_yaml(phpy_yaml, self)
+        set_data_to_phonopy_yaml(phpy_yaml, self)
         return phpy_yaml
 
     ###################
@@ -3528,8 +3524,8 @@ class Phonopy:
         """
         if self._dataset is None:
             raise RuntimeError("Displacement-force dataset is not set.")
-        if target in self._dataset:  # type-2
-            return self._dataset[target]
+        if "displacements" in self._dataset and target in self._dataset:  # type-2
+            return self._dataset[target]  # type: ignore
         if "first_atoms" in self._dataset:  # type-1
             values = []
             for disp in self._dataset["first_atoms"]:
@@ -3545,18 +3541,19 @@ class Phonopy:
 
     def _set_forces_energies(
         self,
-        values: NDArray[np.double]
+        values: Sequence[float]
+        | NDArray[np.double]
         | Sequence[NDArray[np.double]]
         | Sequence[Sequence[Sequence[float]]],
         target: Literal["forces", "supercell_energies"],
     ) -> None:
         assert self._dataset is not None
         if "first_atoms" in self._dataset:  # type-1
-            for disp, v in zip(self._dataset["first_atoms"], values, strict=True):  # type: ignore
+            for disp, val in zip(self._dataset["first_atoms"], values, strict=True):  # type: ignore
                 if target == "forces":
-                    disp[target] = np.array(v, dtype="double", order="C")
+                    disp[target] = np.array(val, dtype="double", order="C")
                 elif target == "supercell_energies":
-                    disp["supercell_energy"] = float(v)
+                    disp["supercell_energy"] = float(val)  # type: ignore
         elif "displacements" in self._dataset:  # type-2
             _values = np.array(values, dtype="double", order="C")
             natom = len(self._supercell)
@@ -3573,7 +3570,7 @@ class Phonopy:
             raise RuntimeError("Set of displacements is not available.")
 
 
-def set_yaml(phpy_yaml: PhonopyYaml, self: Phonopy) -> None:
+def set_data_to_phonopy_yaml(phpy_yaml: PhonopyYaml, self: Phonopy) -> None:
     """Set data to PhonopyYaml instance."""
     phpy_yaml.unitcell = self.unitcell
     phpy_yaml.primitive = self.primitive

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/harmonic/force_constants.py

@@ -41,6 +41,7 @@ from collections.abc import Sequence
 import numpy as np
 from numpy.typing import NDArray
 
+from phonopy.harmonic.displacement import Type1DisplacementDataset
 from phonopy.structure.atoms import PhonopyAtoms
 from phonopy.structure.cells import (
     Primitive,
@@ -64,7 +65,7 @@ class FDFCSolver:
         supercell: PhonopyAtoms,
         primitive: Primitive | None,
         symmetry: Symmetry,
-        dataset: dict,
+        dataset: Type1DisplacementDataset,
         is_compact_fc: bool = False,
         log_level: int = 0,  # currently not used
     ):
@@ -93,7 +94,7 @@ class FDFCSolver:
         self,
         supercell: PhonopyAtoms,
         symmetry: Symmetry,
-        dataset: dict,
+        dataset: Type1DisplacementDataset,
         atom_list: NDArray[np.int64] | None = None,
         primitive: Primitive | None = None,
     ) -> NDArray[np.double]:
@@ -830,7 +831,7 @@ def _solve_force_constants_svd(
 def _get_force_constants_disps(
     force_constants: NDArray[np.double],
     supercell: PhonopyAtoms,
-    dataset: dict,
+    dataset: Type1DisplacementDataset,
     symmetry: Symmetry,
     atom_list: NDArray[np.int64] | None = None,
 ) -> NDArray[np.int64]:
@@ -861,11 +862,12 @@ def _get_force_constants_disps(
         disps = []
         sets_of_forces = []
 
-        for x in dataset["first_atoms"]:
-            if x["number"] != disp_atom_number:
+        for data_1st in dataset["first_atoms"]:
+            if data_1st["number"] != disp_atom_number:
                 continue
-            disps.append(x["displacement"])
-            sets_of_forces.append(x["forces"])
+            disps.append(data_1st["displacement"])
+            assert "forces" in data_1st
+            sets_of_forces.append(data_1st["forces"])
 
         site_symmetry = symmetry.get_site_symmetry(disp_atom_number)
 

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/interface/fc_calculator.py

@@ -46,6 +46,7 @@ from phonopy.exception import (
     ForceCalculatorRequiredError,
     ForceConstantsCalculatorNotFoundError,
 )
+from phonopy.harmonic.displacement import DisplacementDataset
 from phonopy.harmonic.force_constants import FDFCSolver
 from phonopy.interface.alm import ALMFCSolver
 from phonopy.interface.symfc import SymfcFCSolver, parse_symfc_options
@@ -77,7 +78,7 @@ def check_and_cast_fc_calculator_name(
 
 def get_fc_solver(
     supercell: PhonopyAtoms,
-    dataset: dict,
+    dataset: DisplacementDataset,
     primitive: Primitive | None = None,
     fc_calculator: Literal["traditional", "symfc", "alm"] | None = None,
     fc_calculator_options: str | None = None,
@@ -92,7 +93,7 @@ def get_fc_solver(
     ----------
     supercell : PhonopyAtoms
         Supercell
-    dataset : dict
+    dataset : DisplacementDataset
         Dataset that contains displacements, forces, and optionally energies.
     primitive : Primitive
         Primitive cell. Only needed for the traditional FC calculator.
@@ -135,7 +136,7 @@ def get_fc_solver(
 
 def get_fc2(
     supercell: PhonopyAtoms,
-    dataset: dict,
+    dataset: DisplacementDataset,
     primitive: Primitive | None = None,
     fc_calculator: Literal["traditional", "symfc", "alm"] | None = None,
     fc_calculator_options: str | None = None,
@@ -152,7 +153,7 @@ def get_fc2(
     ----------
     supercell : PhonopyAtoms
         Supercell
-    dataset : dict
+    dataset : DisplacementDataset
         Dataset that contains displacements, forces, and optionally energies.
     primitive : Primitive
         Primitive cell. Only needed for the traditional FC calculator.
@@ -199,7 +200,7 @@ class FCSolver:
         fc_solver_name: Literal["traditional", "symfc", "alm"],
         supercell: PhonopyAtoms,
         symmetry: Symmetry | None = None,
-        dataset: dict | None = None,
+        dataset: DisplacementDataset | None = None,
         is_compact_fc: bool = False,
         primitive: Primitive | None = None,
         orders: Sequence[int] | None = None,
@@ -222,7 +223,7 @@ class FCSolver:
             Supercell.
         symmetry : Symmetry, optional
             Symmetry of supercell. Default is None.
-        dataset : dict, optional
+        dataset : DisplacementDataset, optional
             Dataset that contains displacements, forces, and optionally
             energies. Default is None.
         is_compact_fc : bool, optional

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/interface/phonopy_yaml.py

@@ -1213,12 +1213,12 @@ def load_yaml(fp: str | os.PathLike | typing.IO) -> dict:
     lzma and gzip comppressed non-stream files can be loaded.
 
     """
-    if isinstance(fp, io.IOBase):
-        yaml_data = yaml.load(fp, Loader=Loader)
-    else:
+    if isinstance(fp, (str, os.PathLike)):
         myio = get_io_module_to_decompress(fp)
         with myio.open(fp, "rb") as f:
             yaml_data = yaml.load(f, Loader=Loader)
+    else:
+        yaml_data = yaml.load(fp, Loader=Loader)
 
     return yaml_data
 

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/phonon/dos.py

@@ -413,8 +413,8 @@ class ProjectedDos(Dos):
     def plot(
         self,
         ax,
-        indices: list[list[int]] | None = None,
-        legend: list[str] | None = None,
+        indices: Sequence[Sequence[int]] | None,
+        legend: Sequence[str] | None = None,
         legend_prop: dict | None = None,
         legend_frameon: bool = True,
         xlabel: str | None = None,
@@ -615,8 +615,8 @@ def plot_projected_dos(
     ax,
     frequency_points: NDArray[np.double],
     projected_dos: NDArray[np.double],
-    indices: list[list[int]] | None = None,
-    legend: list[str] | None = None,
+    indices: Sequence[Sequence[int]] | None = None,
+    legend: Sequence[str] | None = None,
     legend_prop: dict | None = None,
     legend_frameon: bool = True,
     xlabel: str | None = None,

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/phonon/modulation.py

@@ -65,7 +65,7 @@ class Modulation:
         phonon_modes: Sequence[tuple[NDArray[np.double], int, float, float]],
         delta_q: NDArray[np.double] | Sequence[float] | None = None,
         derivative_order: int | None = None,
-        nac_q_direction: NDArray[np.double] | None = None,
+        nac_q_direction: Sequence[float] | NDArray[np.double] | None = None,
         factor: float | None = None,
     ) -> None:
         """Init method."""
@@ -74,7 +74,11 @@ class Modulation:
         self._phonon_modes = phonon_modes
         self._dimension = np.array(dimension, dtype="int64").ravel()
         self._delta_q = delta_q
-        self._nac_q_direction = nac_q_direction
+        self._nac_q_direction = (
+            np.array(nac_q_direction, dtype="double")
+            if nac_q_direction is not None
+            else None
+        )
         self._ddm = DerivativeOfDynamicalMatrix(dynamical_matrix)
         self._derivative_order = derivative_order
 

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/phonon/moment.py

@@ -54,7 +54,7 @@ class PhononMoment:
     def __init__(
         self,
         frequencies: NDArray[np.double],
-        weights: NDArray[np.double],
+        weights: NDArray[np.int64],
         eigenvectors: NDArray[np.cdouble] | None = None,
     ) -> None:
         """Init method."""

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/phonon/qpoints.py

@@ -93,7 +93,7 @@ class QpointsPhonon:
         | Sequence[NDArray[np.double]]
         | NDArray[np.double],
         dynamical_matrix: DynamicalMatrix | DynamicalMatrixNAC,
-        nac_q_direction: NDArray[np.double] | None = None,
+        nac_q_direction: Sequence[float] | NDArray[np.double] | None = None,
         with_eigenvectors: bool = False,
         group_velocity: GroupVelocity | None = None,
         with_dynamical_matrices: bool = False,
@@ -132,7 +132,11 @@ class QpointsPhonon:
 
         self._qpoints = np.asarray(qpoints, dtype="double", order="C")
         self._dynamical_matrix = dynamical_matrix
-        self._nac_q_direction = nac_q_direction
+        self._nac_q_direction = (
+            np.array(nac_q_direction, dtype="double")
+            if nac_q_direction is not None
+            else None
+        )
         self._with_eigenvectors = with_eigenvectors
         self._gv_obj: GroupVelocity | None = group_velocity
         self._with_dynamical_matrices = with_dynamical_matrices

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/phonon/thermal_displacement.py

@@ -101,7 +101,7 @@ class ThermalMotion:
         return self._temperatures
 
     @temperatures.setter
-    def temperatures(self, temperatures: NDArray[np.double]) -> None:
+    def temperatures(self, temperatures: Sequence[float] | NDArray[np.double]) -> None:
         t_array = np.array(temperatures, dtype="double")
         condition = np.logical_not(t_array < 0)
         self._temperatures = np.extract(condition, t_array).astype("double")

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/spectrum/dynamic_structure_factor.py

@@ -36,7 +36,7 @@
 
 from __future__ import annotations
 
-from collections.abc import Callable
+from collections.abc import Callable, Sequence
 from typing import cast
 
 import numpy as np
@@ -106,7 +106,7 @@ class DynamicStructureFactor:
     def __init__(
         self,
         mesh_phonon: Mesh | IterMesh,
-        Qpoints: NDArray[np.double],
+        Qpoints: Sequence[Sequence[float]] | NDArray[np.double],
         T: float,
         atomic_form_factor_func: Callable[[str, float], float] | None = None,
         scattering_lengths: dict[str, float] | None = None,

{phonopy-3.2.0 → phonopy-3.2.1}/phonopy/version.py

@@ -34,4 +34,4 @@
 # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 # POSSIBILITY OF SUCH DAMAGE.
 
-__version__ = "3.2.0"
+__version__ = "3.2.1"