build(docs): pass mkdocs --strict by adding missing type hints and docstring fixes

Author: skblv

README.md

@@ -316,8 +316,7 @@ If using providers not present in the built-in pricing DB, set both `model_input
 ### Local MkDocs Site
 1. Install the documentation dependencies: `pip install mkdocs mkdocs-material mkdocstrings mkdocstrings-python`
 2. Serve the docs locally to `http://127.0.0.1:8000/`: `mkdocs serve`
-
-Use `mkdocs build` to generate a static site in the `site/` directory.
+3. Use `mkdocs build` to generate a static site in the `site/` directory.
 
 ### Reference Materials
 - [Schema Reference](SCHEMA_REFERENCE.md) - Detailed schema configuration guide

src/delm/config.py

@@ -687,7 +687,9 @@ def from_yaml(cls, path: Path) -> "DELMConfig":
         return cls.from_dict(data)
 
     @staticmethod
-    def from_any(config_like) -> "DELMConfig":
+    def from_any(
+        config_like: "DELMConfig | dict[str, Any] | str | Path",
+    ) -> "DELMConfig":
         """Create ``DELMConfig`` from various input types.
 
         Args:

src/delm/delm.py

@@ -136,7 +136,7 @@ def from_yaml(
         config_path: Union[str, Path],
         experiment_name: str,
         experiment_directory: Path,
-        **kwargs,
+        **kwargs: Any,
     ) -> "DELM":
         """Create a DELM instance from a YAML configuration file.
 
@@ -168,7 +168,7 @@ def from_dict(
         config_dict: Dict[str, Any],
         experiment_name: str,
         experiment_directory: Path,
-        **kwargs,
+        **kwargs: Any,
     ) -> "DELM":
         """Create a DELM instance from a configuration dictionary.
 

src/delm/utils/retry_handler.py

@@ -20,7 +20,7 @@ def __init__(self, max_retries: int = 3, base_delay: float = 1.0):
         self.max_retries = max_retries
         self.base_delay = base_delay
 
-    def execute_with_retry(self, func: Callable, *args, **kwargs) -> Any:
+    def execute_with_retry(self, func: Callable, *args: Any, **kwargs: Any) -> Any:
         """Execute function with retry logic.
         
         Args:
@@ -32,7 +32,7 @@ def execute_with_retry(self, func: Callable, *args, **kwargs) -> Any:
             The result of the function execution.
 
         Raises:
-            The last exception from the function execution if all attempts fail.
+            Exception: The last exception from the function execution if all attempts fail.
         """
         exceptions = []
         for attempt in range(self.max_retries + 1):