MAINT strict mypy checking on the whole repository (microsoft#1310)

Author: romanlutz

.pre-commit-config.yaml

@@ -55,18 +55,11 @@ repos:
     hooks:
       - id: mypy
         name: mypy (strict)
-        files: &strict_modules ^pyrit/(auth|analytics|embedding|exceptions|memory|prompt_normalizer)/
+        files: &strict_modules ^pyrit/
         args: [--install-types, --non-interactive, --ignore-missing-imports, --sqlite-cache, --cache-dir=.mypy_cache, --strict]
         entry: mypy
         language: system
         types: [ python ]
-      - id: mypy
-        name: mypy (regular)
-        exclude: *strict_modules
-        args: [--install-types, --non-interactive, --ignore-missing-imports, --sqlite-cache, --cache-dir=.mypy_cache]
-        entry: mypy
-        language: system
-        types: [ python ]
 
   - repo: local
     hooks:

pyproject.toml

@@ -84,8 +84,17 @@ dev = [
     "ruff>=0.14.4",
     "sphinxcontrib-mermaid>=1.0.0",
     "types-aiofiles>=24.1.0",
+    "types-cachetools>=5.5.0",
+    "types-decorator>=5.1.0",
+    "types-paramiko>=3.5.0",
+    "types-pycurl>=7.45.0",
+    "types-pytz>=2024.2.0",
     "types-PyYAML>=6.0.12.20250516",
     "types-requests>=2.31.0.20250515",
+    "types-simplejson>=3.19.0",
+    "types-six>=1.16.0",
+    "types-tabulate>=0.9.0",
+    "types-ujson>=5.10.0",
 ]
 huggingface = [
     "torch>=2.7.0",

pyrit/analytics/conversation_analytics.py

@@ -90,7 +90,7 @@ def get_similar_chat_messages_by_embedding(
         return similar_messages
 
 
-def cosine_similarity(a: np.ndarray, b: np.ndarray) -> float:
+def cosine_similarity(a: np.ndarray, b: np.ndarray) -> float:  # type: ignore[type-arg, unused-ignore]
     """
     Calculate the cosine similarity between two 1D vectors.
 

pyrit/auth/azure_storage_auth.py

@@ -2,7 +2,6 @@
 # Licensed under the MIT license.
 
 from datetime import datetime, timedelta
-from typing import cast
 from urllib.parse import urlparse
 
 from azure.identity.aio import DefaultAzureCredential
@@ -89,11 +88,11 @@ async def get_sas_token(container_url: str) -> str:
                     account_name=storage_account_name,
                     container_name=container_name,
                     user_delegation_key=user_delegation_key,
-                    permission=ContainerSasPermissions(read=True, write=True, create=True, list=True, delete=True),  # type: ignore
+                    permission=ContainerSasPermissions(read=True, write=True, create=True, list=True, delete=True),  # type: ignore[no-untyped-call, unused-ignore]
                     expiry=expiry_time,
                     start=start_time,
                 )
         finally:
             await credential.close()
 
-        return cast(str, sas_token)
+        return sas_token  # type: ignore[return-value, no-any-return, unused-ignore]