Skip to content

Commit ae9ddfd

Browse files
adamtheturtleclaude
andcommitted
fix: silence pyright 1.1.409 reportPrivateImportUsage on torch.tensor
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
1 parent 5da47e8 commit ae9ddfd

2 files changed

Lines changed: 18 additions & 3 deletions

File tree

src/mock_vws/image_matchers.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,12 @@ def __call__(
8181
second_image_resized = second_image.resize(size=target_size)
8282

8383
first_image_np = np.array(object=first_image_resized, dtype=np.float32)
84-
first_image_tensor = torch.tensor(data=first_image_np).float() / 255
84+
first_image_tensor = (
85+
torch.tensor( # pyright: ignore[reportPrivateImportUsage]
86+
data=first_image_np,
87+
).float()
88+
/ 255
89+
)
8590
first_image_tensor = first_image_tensor.view(
8691
first_image_resized.size[1],
8792
first_image_resized.size[0],
@@ -92,7 +97,12 @@ def __call__(
9297
object=second_image_resized,
9398
dtype=np.float32,
9499
)
95-
second_image_tensor = torch.tensor(data=second_image_np).float() / 255
100+
second_image_tensor = (
101+
torch.tensor( # pyright: ignore[reportPrivateImportUsage]
102+
data=second_image_np,
103+
).float()
104+
/ 255
105+
)
96106
second_image_tensor = second_image_tensor.view(
97107
second_image_resized.size[1],
98108
second_image_resized.size[0],

src/mock_vws/target_raters.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,12 @@ def _get_brisque_target_tracking_rating(*, image_content: bytes) -> int:
2828
image_file = io.BytesIO(initial_bytes=image_content)
2929
with Image.open(fp=image_file) as image:
3030
image_np = np.array(object=image, dtype=np.float32)
31-
image_tensor = torch.tensor(data=image_np).float() / 255
31+
image_tensor = (
32+
torch.tensor( # pyright: ignore[reportPrivateImportUsage]
33+
data=image_np,
34+
).float()
35+
/ 255
36+
)
3237
image_tensor = image_tensor.view(
3338
image.size[1],
3439
image.size[0],

0 commit comments

Comments
 (0)