Small typing improvements

openstreetmap-polska · Jan 5, 2025 · d18a724 · d18a724
1 parent 5ff4db9
commit d18a724
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/country_code_assigner.py b/country_code_assigner.py
@@ -2,7 +2,7 @@ class CountryCodeAssigner:
     __slots__ = 'used'
 
     def __init__(self):
-        self.used = set()
+        self.used: set[str] = set()
 
     def get_unique(self, tags: dict[str, str]) -> str:
         for check_used in (True, False):

diff --git a/openstreetmap.py b/openstreetmap.py
@@ -16,7 +16,7 @@ def osm_user_has_active_block(user: dict) -> bool:
 
 class OpenStreetMap:
     def __init__(self, access_token: SecretStr):
-        self.access_token = access_token
+        self.access_token: SecretStr = access_token
 
     @retry_exponential(10)
     @trace

diff --git a/xmltodict_postprocessor.py b/xmltodict_postprocessor.py
@@ -1,8 +1,8 @@
-def xmltodict_postprocessor(_, key, value):
-    if key in ('@id', '@ref', '@changeset', '@uid'):
+def xmltodict_postprocessor(_, key: str, value: str):
+    if key in {'@id', '@ref', '@changeset', '@uid'}:
         return key, int(value)
 
-    if key in ('@lon, @lat'):
+    if key in {'@lon', '@lat'}:
         return key, float(value)
 
     if key == '@version':