diff options
Diffstat (limited to 'src/pyssg/yaml_parser.py')
-rw-r--r-- | src/pyssg/yaml_parser.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/pyssg/yaml_parser.py b/src/pyssg/yaml_parser.py index 2e1548b..aeb164e 100644 --- a/src/pyssg/yaml_parser.py +++ b/src/pyssg/yaml_parser.py @@ -3,6 +3,7 @@ from yaml import SafeLoader from yaml.nodes import SequenceNode from importlib.resources import path as rpath from logging import Logger, getLogger +from typing import Any log: Logger = getLogger(__name__) @@ -17,15 +18,15 @@ def setup_custom_yaml() -> None: SafeLoader.add_constructor('!join', __join_constructor) -def __read_raw_yaml(path: str) -> list[dict]: - all_docs: list[dict] = [] +def __read_raw_yaml(path: str) -> list[dict[str, Any]]: + all_docs: list[dict[str, Any]] = [] with open(path, 'r') as f: for doc in yaml.safe_load_all(f): all_docs.append(doc) return all_docs -def get_parsed_yaml(resource: str, package: str = '') -> list[dict]: +def get_parsed_yaml(resource: str, package: str = '') -> list[dict[str, Any]]: if package == '': log.debug('parsing yaml; reading "%s"', resource) return __read_raw_yaml(resource) |