Skip to content

Commit 14d7de1

Browse files
authored
Fix type annotations (#5166)
1 parent c9b420a commit 14d7de1

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/sagemaker/feature_store/feature_processor/_input_offset_parser.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -72,14 +72,16 @@ def get_offset_datetime(self, offset: Optional[str]) -> datetime:
7272

7373
return self.now + offset_td
7474

75-
def get_offset_date_year_month_day_hour(self, offset: Optional[str]) -> Tuple[str]:
75+
def get_offset_date_year_month_day_hour(
76+
self, offset: Optional[str]
77+
) -> Tuple[str, str, str, str]:
7678
"""Get the year, month, day and hour based on offset diff.
7779
7880
Args:
7981
offset (Optional[str]): Offset that is used for target date calcluation.
8082
8183
Returns:
82-
Tuple[str]: A tuple that consists of extracted year, month, day, hour from offset date.
84+
Tuple[str, str, str, str]: A tuple that consists of extracted year, month, day, hour from offset date.
8385
"""
8486
if offset is None:
8587
return (None, None, None, None)

0 commit comments

Comments
 (0)