diff --git a/src/waybar_check_gmail/config/config.py b/src/waybar_check_gmail/config/config.py index 056af9e..ede2c24 100644 --- a/src/waybar_check_gmail/config/config.py +++ b/src/waybar_check_gmail/config/config.py @@ -75,8 +75,8 @@ def __init__(self, fn: AnyPath): else: self._map[sec][key] = val - def section(self, section): + def section(self, section: str) -> Dict[str, Union[str, bool, None]]: return self._map[section] - def get(self, section, key): + def get(self, section: str, key: str) -> Union[str, bool, None]: return self._map[section][key]