Merge pull request #2461 from VWS-Python/follow-imports

Follow imports on mypy

Author: adamtheturtle

pyproject.toml

@@ -64,7 +64,7 @@ optional-dependencies.dev = [
     "freezegun==1.5.1",
     "furo==2024.8.6",
     "interrogate==1.7.0",
-    "mypy[faster-cache]==1.13.0",
+    "mypy[faster-cache]==1.14.0",
     "pre-commit==4.0.1",
     "pydocstyle==6.3",
     "pyenchant==3.3.0rc1",
@@ -384,6 +384,7 @@ exclude = [ "build" ]
 plugins = [
     "pydantic.mypy",
 ]
+follow_untyped_imports = true
 
 [tool.pyright]
 reportUnnecessaryTypeIgnoreComment = true

src/mock_vws/target_raters.py

@@ -9,10 +9,10 @@
 from typing import Protocol, runtime_checkable
 
 import numpy as np
-import piq  # type: ignore[import-untyped]
 import torch
 from beartype import beartype
 from PIL import Image
+from piq.brisque import brisque  # pyright: ignore[reportMissingTypeStubs]
 
 
 @functools.cache
@@ -38,7 +38,7 @@ def _get_brisque_target_tracking_rating(*, image_content: bytes) -> int:
     )
     image_tensor = image_tensor.permute(2, 0, 1).unsqueeze(dim=0)
     try:
-        brisque_score = piq.brisque(x=image_tensor, data_range=255)
+        brisque_score = brisque(x=image_tensor, data_range=255)
     except (AssertionError, IndexError):
         return 0
     return math.ceil(int(brisque_score.item()) / 20)