Skip to content

Commit

Permalink
fix pylint
Browse files Browse the repository at this point in the history
  • Loading branch information
yichao-liang committed Dec 26, 2024
1 parent 1fe5915 commit 06fd4af
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion predicators/image_patch_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,7 @@ def __init__(
# (H, W, C)
# Convert to shape (C, H, W)
img = np.transpose(img, (2, 0, 1))
image_tensor = th.tensor(
image_tensor = th.tensor( # pylint: disable=no-member
img,
dtype=th.float32 # pylint: disable=no-member
) / 255.0
Expand Down

0 comments on commit 06fd4af

Please sign in to comment.