diff --git a/.github/scripts/benchmarks/gather_runners_info.py b/.github/scripts/benchmarks/gather_runners_info.py index 14bc6f4fe0..d5218218d6 100755 --- a/.github/scripts/benchmarks/gather_runners_info.py +++ b/.github/scripts/benchmarks/gather_runners_info.py @@ -13,7 +13,7 @@ from logging import info from typing import Any, Dict -import psutil +import psutil # type: ignore[import-untyped] logging.basicConfig(level=logging.INFO) @@ -32,7 +32,7 @@ def get_runner_info() -> Dict[str, Any]: device_type = "" try: - import torch + import torch # type: ignore[import-not-found] if torch.cuda.is_available(): if torch.version.hip: diff --git a/.lintrunner.toml b/.lintrunner.toml index a8aae82569..2a4a36ecb6 100644 --- a/.lintrunner.toml +++ b/.lintrunner.toml @@ -38,7 +38,7 @@ init_command = [ [[linter]] code = 'MYPY' include_patterns = [ - '.github/scripts/*.py', + '.github/scripts/**/*.py', 'aws/lambda/**/.py', 'stats/**/*.py', 'stats/**/*.pyi', @@ -46,7 +46,7 @@ include_patterns = [ 'tools/**/*.pyi', 'torchci/**/*.py', 'torchci/**/*.pyi', - 's3_management/*.py', + 's3_management/**/*.py', ] exclude_patterns = [ 'aws/lambda/servicelab-ingestor/**',