add mypy lintrunner support

Signed-off-by: Eli Uriegas <eliuriegas@meta.com>

Author: seemethere

.lintrunner.toml

@@ -46,6 +46,7 @@ include_patterns = [
     'tools/**/*.pyi',
     'torchci/**/*.py',
     'torchci/**/*.pyi',
+    's3_management/*.py',
 ]
 exclude_patterns = [
     'aws/lambda/servicelab-ingestor/**',

s3_management/manage.py

@@ -11,9 +11,9 @@
 from datetime import datetime
 from os import makedirs, path
 from re import match, search, sub
-from typing import Dict, Iterable, List, Optional, Set, Type, TypeVar
+from typing import Dict, Iterable, List, Optional, Set, Type, TypeVar, Union
 
-import boto3
+import boto3  # type: ignore[import]
 import botocore  # type: ignore[import]
 from packaging.version import (
     InvalidVersion,
@@ -243,7 +243,7 @@ def __lt__(self, other):
         return self.key < other.key
 
 
-def safe_parse_version(ver_str: str) -> LegacyVersion | Version:
+def safe_parse_version(ver_str: str) -> Union[LegacyVersion, Version]:
     try:
         return _parse_version(ver_str)
     except InvalidVersion: