diff --git a/geonamescache/mappers.py b/geonamescache/mappers.py index 3c02c4a..45d1aa4 100644 --- a/geonamescache/mappers.py +++ b/geonamescache/mappers.py @@ -4,7 +4,8 @@ from geonamescache import GeonamesCache from . import mappings -from .types import ContinentCode, CountryFields, CountryNumericFields +from .types import (ContinentCode, CountryFields, CountryNumericFields, + CountryStringFields) @overload @@ -16,20 +17,7 @@ def country( @overload def country( from_key: str = "name", - to_key: Literal[ - "capital", - "currencycode", - "currencyname", - "iso", - "iso3", - "fips", - "languages", - "name", - "neighbours", - "phone", - "postalcoderegex", - "tld", - ] = "iso", + to_key: CountryStringFields = "iso", ) -> Callable[[str], str]: ...