Add type annotations for io.vasp.inputs/optics

dev_scripts/update_pt_data.py

@@ -234,7 +234,7 @@ def gen_iupac_ordering():
 def add_electron_affinities():
     """Update the periodic table data file with electron affinities."""
 
-    req = requests.get("https://wikipedia.org/wiki/Electron_affinity_(data_page)")
+    req = requests.get("https://wikipedia.org/wiki/Electron_affinity_(data_page)", timeout=60)
     soup = BeautifulSoup(req.text, "html.parser")
     table = None
     for table in soup.find_all("table"):

pymatgen/alchemy/filters.py

@@ -244,7 +244,7 @@ def __init__(self, existing_structures, structure_matcher=None, symprec=None):
                 structure matcher is used. A recommended value is 1e-5.
         """
         self.symprec = symprec
-        self.structure_list = []
+        self.structure_list: list = []
         self.existing_structures = existing_structures
         if isinstance(structure_matcher, dict):
             self.structure_matcher = StructureMatcher.from_dict(structure_matcher)

pymatgen/alchemy/transmuters.py

@@ -244,7 +244,7 @@ def __init__(self, cif_string, transformations=None, primitive=True, extend_coll
         """
         transformed_structures = []
         lines = cif_string.split("\n")
-        structure_data = []
+        structure_data: list = []
         read_data = False
         for line in lines:
             if re.match(r"^\s*data", line):

pymatgen/analysis/chemenv/coordination_environments/coordination_geometries.py

@@ -908,7 +908,7 @@ def __init__(self, permutations_safe_override=False, only_symbols=None):
 
         self.minpoints = {}
         self.maxpoints = {}
-        self.separations_cg = {}
+        self.separations_cg: dict = {}
         for cn in range(6, 21):
             for cg in self.get_implemented_geometries(coordination=cn):
                 if only_symbols is not None and cg.ce_symbol not in only_symbols:

pymatgen/analysis/chemenv/coordination_environments/structure_environments.py

@@ -2085,7 +2085,7 @@ def __init__(self, coord_geoms=None):
             coord_geoms: coordination geometries to be added to the chemical environment.
         """
         if coord_geoms is None:
-            self.coord_geoms = {}
+            self.coord_geoms: dict = {}
         else:
             raise NotImplementedError(
                 "Constructor for ChemicalEnvironments with the coord_geoms argument is not yet implemented"

pymatgen/analysis/ewald.py

@@ -540,7 +540,7 @@ def __init__(self, matrix, m_list, num_to_return=1, algo=ALGO_FAST):
         if algo == EwaldMinimizer.ALGO_COMPLETE:
             raise NotImplementedError("Complete algo not yet implemented for EwaldMinimizer")
 
-        self._output_lists = []
+        self._output_lists: list = []
         # Tag that the recurse function looks at each level. If a method
         # sets this to true it breaks the recursion and stops the search.
         self._finished = False

pymatgen/analysis/structure_prediction/substitution_probability.py

@@ -79,7 +79,7 @@ def __init__(self, lambda_table=None, alpha=-5):
 
         # create Z and px
         self.Z = 0
-        self._px = defaultdict(float)
+        self._px: defaultdict = defaultdict(float)
         for s1, s2 in itertools.product(self.species, repeat=2):
             value = math.exp(self.get_lambda(s1, s2))
             self._px[s1] += value / 2

pymatgen/analysis/wulff.py

@@ -90,8 +90,8 @@ def __init__(self, normal, e_surf, normal_pt, dual_pt, index, m_ind_orig, miller
         self.index = index
         self.m_ind_orig = m_ind_orig
         self.miller = miller
-        self.points = []
-        self.outer_lines = []
+        self.points: list = []
+        self.outer_lines: list = []
 
 
 class WulffShape:

pymatgen/apps/battery/plotter.py

@@ -28,7 +28,7 @@ def __init__(self, xaxis="capacity", hide_negative=False):
             - frac_x: the atomic fraction of the working ion
             hide_negative: If True only plot the voltage steps above zero.
         """
-        self._electrodes = {}
+        self._electrodes: dict = {}
         self.xaxis = xaxis
         self.hide_negative = hide_negative
 

pymatgen/apps/borg/queen.py

@@ -44,7 +44,7 @@ def __init__(self, drone, rootpath=None, number_of_drones=1):
         """
         self._drone = drone
         self._num_drones = number_of_drones
-        self._data = []
+        self._data: list = []
 
         if rootpath:
             if number_of_drones > 1:

pymatgen/command_line/vampire_caller.py

@@ -131,8 +131,8 @@ def __init__(
 
         # Call Vampire
         with subprocess.Popen(["vampire-serial"], stdout=subprocess.PIPE, stderr=subprocess.PIPE) as process:
-            stdout, stderr = process.communicate()
-            stdout = stdout.decode()
+            _stdout, stderr = process.communicate()
+            stdout: str = _stdout.decode()
 
         if stderr:
             van_helsing = stderr.decode()

pymatgen/core/interface.py

@@ -244,7 +244,7 @@ def coincidents(self) -> list[Site]:
                 coincident_sites.append(self.sites[idx])
         return coincident_sites
 
-    def __str__(self):
+    def __str__(self) -> str:
         comp = self.composition
         outs = [
             f"Gb Summary ({comp.formula})",

pymatgen/core/structure.py

@@ -2860,7 +2860,7 @@ def from_str(  # type: ignore[override]
         elif fmt_low == "poscar":
             from pymatgen.io.vasp import Poscar
 
-            struct = Poscar.from_str(input_string, default_names=False, read_velocities=False, **kwargs).structure
+            struct = Poscar.from_str(input_string, default_names=None, read_velocities=False, **kwargs).structure
         elif fmt_low == "cssr":
             from pymatgen.io.cssr import Cssr
 

pymatgen/ext/cod.py

@@ -90,7 +90,7 @@ def get_structure_by_id(self, cod_id, **kwargs):
         Returns:
             A Structure.
         """
-        response = requests.get(f"http://{self.url}/cod/{cod_id}.cif")
+        response = requests.get(f"http://{self.url}/cod/{cod_id}.cif", timeout=60)
         return Structure.from_str(response.text, fmt="cif", **kwargs)
 
     @requires(which("mysql"), "mysql must be installed to use this query.")
@@ -112,7 +112,7 @@ def get_structure_by_formula(self, formula: str, **kwargs) -> list[dict[str, str
         for line in text:
             if line.strip():
                 cod_id, sg = line.split("\t")
-                response = requests.get(f"http://www.crystallography.net/cod/{cod_id.strip()}.cif")
+                response = requests.get(f"http://www.crystallography.net/cod/{cod_id.strip()}.cif", timeout=60)
                 try:
                     struct = Structure.from_str(response.text, fmt="cif", **kwargs)
                     structures.append({"structure": struct, "cod_id": int(cod_id), "sg": sg})

pymatgen/ext/matproj_legacy.py

@@ -1570,7 +1570,7 @@ def _check_get_download_info_url_by_task_id(self, prefix, task_ids) -> list[str]
 
     @staticmethod
     def _check_nomad_exist(url) -> bool:
-        response = requests.get(url=url)
+        response = requests.get(url=url, timeout=60)
         if response.status_code != 200:
             return False
         content = json.loads(response.text)

pymatgen/io/abinit/abitimer.py

@@ -85,11 +85,11 @@ def walk(cls, top=".", ext=".abo"):
     def __init__(self):
         """Initialize object."""
         # List of files that have been parsed.
-        self._filenames = []
+        self._filenames: list = []
 
         # timers[filename][mpi_rank]
         # contains the timer extracted from the file filename associated to the MPI rank mpi_rank.
-        self._timers = {}
+        self._timers: dict = {}
 
     def __iter__(self):
         return iter(self._timers)

pymatgen/io/abinit/pseudos.py

@@ -1033,10 +1033,10 @@ class PseudoParser:
 
     def __init__(self):
         # List of files that have been parsed successfully.
-        self._parsed_paths = []
+        self._parsed_paths: list = []
 
         # List of files that could not been parsed.
-        self._wrong_paths = []
+        self._wrong_paths: list = []
 
     def scan_directory(self, dirname, exclude_exts=(), exclude_fnames=()):
         """
@@ -1228,14 +1228,14 @@ def __init__(self, filepath):
         # In this way, we know that only the first two bound states (with f and n attributes)
         # should be used for constructing an initial guess for the wave functions.
 
-        self.valence_states = {}
+        self.valence_states: dict = {}
         for node in root.find("valence_states"):
             attrib = AttrDict(node.attrib)
             assert attrib.id not in self.valence_states
             self.valence_states[attrib.id] = attrib
 
         # Parse the radial grids
-        self.rad_grids = {}
+        self.rad_grids: dict = {}
         for node in root.findall("radial_grid"):
             grid_params = node.attrib
             gid = grid_params["id"]

pymatgen/io/cp2k/outputs.py

@@ -57,15 +57,15 @@ def __init__(self, filename, verbose=False, auto_load=False):
         # IO Info
         self.filename = filename
         self.dir = os.path.dirname(filename)
-        self.filenames = {}
+        self.filenames: dict = {}
         self.parse_files()
-        self.data = {}
+        self.data: dict = {}
 
         # Material properties/results
         self.input = self.initial_structure = self.lattice = self.final_structure = self.composition = None
         self.efermi = self.vbm = self.cbm = self.band_gap = None
-        self.structures = []
-        self.ionic_steps = []
+        self.structures: list = []
+        self.ionic_steps: list = []
 
         # parse the basic run parameters always
         self.parse_cp2k_params()

pymatgen/io/pwscf.py

@@ -522,7 +522,7 @@ def __init__(self, filename):
             filename (str): Filename.
         """
         self.filename = filename
-        self.data = defaultdict(list)
+        self.data: defaultdict = defaultdict(list)
         self.read_pattern(PWOutput.patterns)
         for k, v in self.data.items():
             if k == "energies":

pymatgen/io/qchem/inputs.py

@@ -243,8 +243,8 @@ def get_str(self) -> str:
         """Return a string representation of an entire input file."""
         return str(self)
 
-    def __str__(self):
-        combined_list = []
+    def __str__(self) -> str:
+        combined_list: list = []
         # molecule section
         combined_list.extend((self.molecule_template(self.molecule), "", self.rem_template(self.rem), ""))
         # opt section

pymatgen/io/vasp/help.py

@@ -11,11 +11,11 @@
 class VaspDoc:
     """A VASP documentation helper."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         """Init for VaspDoc."""
         self.url_template = "http://www.vasp.at/wiki/index.php/%s"
 
-    def print_help(self, tag):
+    def print_help(self, tag: str) -> None:
         """
         Print the help for a TAG.
 
@@ -24,7 +24,7 @@ def print_help(self, tag):
         """
         print(self.get_help(tag))
 
-    def print_jupyter_help(self, tag):
+    def print_jupyter_help(self, tag: str) -> None:
         """
         Display HTML help in ipython notebook.
 
@@ -37,7 +37,7 @@ def print_jupyter_help(self, tag):
         display(HTML(html_str))
 
     @classmethod
-    def get_help(cls, tag, fmt="text"):
+    def get_help(cls, tag: str, fmt: str = "text") -> str:
         """
         Get help on a VASP tag.
 
@@ -48,7 +48,7 @@ def get_help(cls, tag, fmt="text"):
             Help text.
         """
         tag = tag.upper()
-        response = requests.get(f"https://www.vasp.at/wiki/index.php/{tag}", verify=False)
+        response = requests.get(f"https://www.vasp.at/wiki/index.php/{tag}", verify=False, timeout=60)
         soup = BeautifulSoup(response.text)
         main_doc = soup.find(id="mw-content-text")
         if fmt == "text":
@@ -60,14 +60,14 @@ def get_help(cls, tag, fmt="text"):
         return output
 
     @classmethod
-    def get_incar_tags(cls):
+    def get_incar_tags(cls) -> list[str]:
         """Returns: All incar tags."""
         tags = []
         for page in [
             "https://www.vasp.at/wiki/index.php/Category:INCAR",
             "https://www.vasp.at/wiki/index.php?title=Category:INCAR&pagefrom=ML+FF+LCONF+DISCARD#mw-pages",
         ]:
-            response = requests.get(page, verify=False)
+            response = requests.get(page, verify=False, timeout=60)
             soup = BeautifulSoup(response.text)
             for div in soup.findAll("div", {"class": "mw-category-group"}):
                 children = div.findChildren("li")