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..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,14 +72,16 @@ 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)