diff --git a/aws_lambda_powertools/metrics/metrics.py b/aws_lambda_powertools/metrics/metrics.py
index 873f09c6377..c8c16e8bd51 100644
--- a/aws_lambda_powertools/metrics/metrics.py
+++ b/aws_lambda_powertools/metrics/metrics.py
@@ -6,6 +6,7 @@
 from aws_lambda_powertools.metrics.provider.cloudwatch_emf.cloudwatch import AmazonCloudWatchEMFProvider
 
 if TYPE_CHECKING:
+    import datetime
     from collections.abc import Callable
 
     from aws_lambda_powertools.metrics.base import MetricResolution, MetricUnit
@@ -133,7 +134,7 @@ def serialize_metric_set(
     def add_metadata(self, key: str, value: Any) -> None:
         self.provider.add_metadata(key=key, value=value)
 
-    def set_timestamp(self, timestamp: int):
+    def set_timestamp(self, timestamp: int | datetime.datetime):
         """
         Set the timestamp for the metric.