From e1b9baccb7424fab2f2e3862add9497fd4f77916 Mon Sep 17 00:00:00 2001
From: Leandro Damascena <lcdama@amazon.pt>
Date: Tue, 15 Apr 2025 17:40:29 +0100
Subject: [PATCH] Using generics types

---
 aws_lambda_powertools/tracing/base.py                 | 3 ++-
 aws_lambda_powertools/tracing/tracer.py               | 3 ++-
 tests/functional/tracer/_aws_xray_sdk/test_tracing.py | 2 ++
 tests/unit/test_tracing.py                            | 2 ++
 4 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/aws_lambda_powertools/tracing/base.py b/aws_lambda_powertools/tracing/base.py
index e095287ce62..8469c075222 100644
--- a/aws_lambda_powertools/tracing/base.py
+++ b/aws_lambda_powertools/tracing/base.py
@@ -8,11 +8,12 @@
 
 import abc
 from contextlib import contextmanager
-from typing import TYPE_CHECKING, Any, Generator, Sequence
+from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
     import numbers
     import traceback
+    from collections.abc import Generator, Sequence
 
 
 class BaseSegment(abc.ABC):
diff --git a/aws_lambda_powertools/tracing/tracer.py b/aws_lambda_powertools/tracing/tracer.py
index 503a8e71141..e2de4c66f48 100644
--- a/aws_lambda_powertools/tracing/tracer.py
+++ b/aws_lambda_powertools/tracing/tracer.py
@@ -6,7 +6,7 @@
 import inspect
 import logging
 import os
-from typing import TYPE_CHECKING, Any, Callable, Sequence, TypeVar, cast, overload
+from typing import TYPE_CHECKING, Any, TypeVar, cast, overload
 
 from aws_lambda_powertools.shared import constants
 from aws_lambda_powertools.shared.functions import (
@@ -19,6 +19,7 @@
 
 if TYPE_CHECKING:
     import numbers
+    from collections.abc import Callable, Sequence
 
     from aws_lambda_powertools.tracing.base import BaseProvider, BaseSegment
 
diff --git a/tests/functional/tracer/_aws_xray_sdk/test_tracing.py b/tests/functional/tracer/_aws_xray_sdk/test_tracing.py
index 5f48b233d91..462da7106ab 100644
--- a/tests/functional/tracer/_aws_xray_sdk/test_tracing.py
+++ b/tests/functional/tracer/_aws_xray_sdk/test_tracing.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import contextlib
 
 import pytest
diff --git a/tests/unit/test_tracing.py b/tests/unit/test_tracing.py
index 425230380ea..beb0f36c0ea 100644
--- a/tests/unit/test_tracing.py
+++ b/tests/unit/test_tracing.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import contextlib
 from typing import NamedTuple
 from unittest import mock