Skip to content

Commit a1b7406

Browse files
committed
Use torchvision in a different way to avoid type errors
1 parent b47c313 commit a1b7406

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/mock_vws/image_matchers.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from torchmetrics.image import (
88
StructuralSimilarityIndexMeasure,
99
)
10-
from torchvision.transforms import functional # type: ignore[import-untyped]
10+
from torchvision import transforms # type: ignore[import-untyped]
1111

1212

1313
@runtime_checkable
@@ -74,8 +74,10 @@ def __call__(
7474
first_image = first_image.resize(size=target_size)
7575
second_image = second_image.resize(size=target_size)
7676

77-
first_image_tensor = functional.to_tensor(pic=first_image) # pyright: ignore[reportUnknownMemberType]
78-
second_image_tensor = functional.to_tensor(pic=second_image) # pyright: ignore[reportUnknownMemberType]
77+
transform = transforms.ToTensor()
78+
79+
first_image_tensor = transform(pic=first_image)
80+
second_image_tensor = transform(pic=second_image)
7981

8082
first_image_tensor_batch_dimension = first_image_tensor.unsqueeze(0)
8183
second_image_tensor_batch_dimension = second_image_tensor.unsqueeze(0)

0 commit comments

Comments
 (0)