various species object improvements

Author: johannes-moegerle

src/rydstate/species/species_object.py

@@ -4,16 +4,24 @@
 import logging
 from abc import ABC
 from functools import cache, cached_property
-from typing import TYPE_CHECKING, ClassVar, overload
+from typing import TYPE_CHECKING, ClassVar, TypeVar, overload
 
 from rydstate.units import rydberg_constant, ureg
 
 if TYPE_CHECKING:
-    from typing_extensions import Self
+    from collections.abc import Callable
+
+    from typing_extensions import ParamSpec, Self
 
     from rydstate.radial.model import PotentialType
     from rydstate.units import PintFloat
 
+    P = ParamSpec("P")
+    R = TypeVar("R")
+
+    def cache(func: Callable[P, R]) -> Callable[P, R]: ...  # type: ignore [misc]
+
+
 logger = logging.getLogger(__name__)
 
 

src/rydstate/species/sqdt/species_object_sqdt.py

@@ -71,10 +71,6 @@ def _setup_nist_energy_levels(self, file: Path) -> None:  # noqa: C901, PLR0912
 
         Args:
             file: Path to the NIST energy levels file.
-            n_max: Maximum principal quantum number for which to load the NIST energy levels.
-                For large quantum numbers, the NIST data is not accurate enough
-                (it does not even show fine structure splitting),
-                so we limit the maximum principal quantum number to 15 by default.
 
         """
         if not file.exists():