fix types from v2 to v3

Author: sthulb

aws_lambda_powertools/logging/formatter.py

@@ -57,7 +57,7 @@ def append_thread_local_keys(self, **additional_keys) -> None:
     def get_current_keys(self) -> dict[str, Any]:
         return {}
 
-    def get_current_thread_keys(self) -> Dict[str, Any]:
+    def get_current_thread_keys(self) -> dict[str, Any]:
         return {}
 
     def remove_keys(self, keys: Iterable[str]) -> None:
@@ -256,7 +256,7 @@ def append_thread_local_keys(self, **additional_keys) -> None:
     def get_current_keys(self) -> dict[str, Any]:
         return self.log_format
 
-    def get_current_thread_keys(self) -> Dict[str, Any]:
+    def get_current_thread_keys(self) -> dict[str, Any]:
         return _get_context().get()
 
     def remove_keys(self, keys: Iterable[str]) -> None:
@@ -371,7 +371,7 @@ def _extract_log_keys(self, log_record: logging.LogRecord) -> dict[str, Any]:
         record_dict["asctime"] = self.formatTime(record=log_record)
         extras = {k: v for k, v in record_dict.items() if k not in RESERVED_LOG_ATTRS}
 
-        formatted_log: Dict[str, Any] = {}
+        formatted_log: dict[str, Any] = {}
 
         # Iterate over a default or existing log structure
         # then replace any std log attribute e.g. '%(level)s' to 'INFO', '%(process)d to '4773'
@@ -428,7 +428,7 @@ def clear_context_keys() -> None:
     _get_context().set({})
 
 
-def set_context_keys(**kwargs: Dict[str, Any]) -> None:
+def set_context_keys(**kwargs: dict[str, Any]) -> None:
     context = _get_context()
     context.set({**context.get(), **kwargs})
 

aws_lambda_powertools/logging/logger.py

@@ -589,7 +589,7 @@ def append_thread_local_keys(self, **additional_keys: object) -> None:
     def get_current_keys(self) -> dict[str, Any]:
         return self.registered_formatter.get_current_keys()
 
-    def get_current_thread_keys(self) -> Dict[str, Any]:
+    def get_current_thread_keys(self) -> dict[str, Any]:
         return self.registered_formatter.get_current_thread_keys()
 
     def remove_keys(self, keys: Iterable[str]) -> None: