diff --git a/pyproject.toml b/pyproject.toml index 444fbf0dd..d55eb9735 100644 --- a/pyproject.toml +++ b/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 diff --git a/src/mock_vws/target_raters.py b/src/mock_vws/target_raters.py index 40f877aa6..8e2cb1d65 100644 --- a/src/mock_vws/target_raters.py +++ b/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)