From 06fd4afd8808875efdc4e7c81cfc88b65a8b06fb Mon Sep 17 00:00:00 2001 From: yichao-liang Date: Thu, 26 Dec 2024 20:14:19 +0000 Subject: [PATCH] fix pylint --- predicators/image_patch_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predicators/image_patch_wrapper.py b/predicators/image_patch_wrapper.py index 85e4feeef..06435ce1c 100644 --- a/predicators/image_patch_wrapper.py +++ b/predicators/image_patch_wrapper.py @@ -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