Skip to content

Commit 1fa22b7

Browse files
committed
Use new style in a new place
1 parent a1b7406 commit 1fa22b7

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/mock_vws/target_raters.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
import piq # type: ignore[import-untyped]
1010
from PIL import Image
11-
from torchvision.transforms import functional # type: ignore[import-untyped]
11+
from torchvision import transforms
1212

1313

1414
@functools.cache
@@ -25,7 +25,8 @@ def _get_brisque_target_tracking_rating(image_content: bytes) -> int:
2525
"""
2626
image_file = io.BytesIO(initial_bytes=image_content)
2727
image = Image.open(fp=image_file)
28-
image_tensor = functional.to_tensor(pic=image) * 255 # pyright: ignore[reportUnknownMemberType]
28+
transform = transforms.ToTensor()
29+
image_tensor = transform(pic=image)
2930
image_tensor = image_tensor.unsqueeze(0)
3031
try:
3132
brisque_score = piq.brisque(x=image_tensor, data_range=255)

0 commit comments

Comments
 (0)