drbh commited on
Commit
5ddd783
·
1 Parent(s): 63f348a

fix: adjust warning

Browse files
Files changed (1) hide show
  1. test_kernel.py +1 -1
test_kernel.py CHANGED
@@ -25,5 +25,5 @@ gray_tensor = img2gray_lib.img2gray(img_tensor).squeeze()
25
  print(gray_tensor.shape) # HW
26
 
27
  # save the output image
28
- gray_img = Image.fromarray(gray_tensor.cpu().numpy().astype(np.uint8), mode="L")
29
  gray_img.save("kernel-builder-logo-gray2.png")
 
25
  print(gray_tensor.shape) # HW
26
 
27
  # save the output image
28
+ gray_img = Image.fromarray(gray_tensor.cpu().numpy().astype(np.uint8))
29
  gray_img.save("kernel-builder-logo-gray2.png")