From 2707706609cb80ac9599fffd19d72e6a7472441b Mon Sep 17 00:00:00 2001
From: Roman A <121314722+GameRoMan@users.noreply.github.com>
Date: Fri, 9 May 2025 18:27:29 +0100
Subject: [PATCH 1/2] Fix type annotations

---
 .../feature_store/feature_processor/_input_offset_parser.py   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/sagemaker/feature_store/feature_processor/_input_offset_parser.py b/src/sagemaker/feature_store/feature_processor/_input_offset_parser.py
index 17e4139bc6..fbd5b65742 100644
--- a/src/sagemaker/feature_store/feature_processor/_input_offset_parser.py
+++ b/src/sagemaker/feature_store/feature_processor/_input_offset_parser.py
@@ -72,14 +72,14 @@ def get_offset_datetime(self, offset: Optional[str]) -> datetime:
 
         return self.now + offset_td
 
-    def get_offset_date_year_month_day_hour(self, offset: Optional[str]) -> Tuple[str]:
+    def get_offset_date_year_month_day_hour(self, offset: Optional[str]) -> Tuple[str, str, str, str]:
         """Get the year, month, day and hour based on offset diff.
 
         Args:
             offset (Optional[str]): Offset that is used for target date calcluation.
 
         Returns:
-            Tuple[str]: A tuple that consists of extracted year, month, day, hour from offset date.
+            Tuple[str, str, str, str]: A tuple that consists of extracted year, month, day, hour from offset date.
         """
         if offset is None:
             return (None, None, None, None)

From 3fa94f23707eda073acde80c8b6ab0b53397d687 Mon Sep 17 00:00:00 2001
From: Roman A <121314722+GameRoMan@users.noreply.github.com>
Date: Fri, 9 May 2025 19:40:40 +0100
Subject: [PATCH 2/2] Correct formatting

---
 .../feature_store/feature_processor/_input_offset_parser.py   | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/sagemaker/feature_store/feature_processor/_input_offset_parser.py b/src/sagemaker/feature_store/feature_processor/_input_offset_parser.py
index fbd5b65742..2b66553ab3 100644
--- a/src/sagemaker/feature_store/feature_processor/_input_offset_parser.py
+++ b/src/sagemaker/feature_store/feature_processor/_input_offset_parser.py
@@ -72,7 +72,9 @@ def get_offset_datetime(self, offset: Optional[str]) -> datetime:
 
         return self.now + offset_td
 
-    def get_offset_date_year_month_day_hour(self, offset: Optional[str]) -> Tuple[str, str, str, str]:
+    def get_offset_date_year_month_day_hour(
+        self, offset: Optional[str]
+    ) -> Tuple[str, str, str, str]:
         """Get the year, month, day and hour based on offset diff.
 
         Args: