No other transform is giving me an error but using the GaussianBlur transform gives me this error:
 File "/home/kaustubh/test/basic_augment.py", line 41, in <module>
    main()
  File "/home/kaustubh/test/basic_augment.py", line 34, in main
    transformed_image = data_transforms(image)
  File "/home/kaustubh/Envs/ml/lib/python3.9/site-packages/torchvision/transforms/transforms.py", line 60, in __call__
    img = t(img)
  File "/home/kaustubh/Envs/ml/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
    return forward_call(*input, **kwargs)
  File "/home/kaustubh/Envs/ml/lib/python3.9/site-packages/torchvision/transforms/transforms.py", line 1730, in forward
    return F.gaussian_blur(img, self.kernel_size, [sigma, sigma])
  File "/home/kaustubh/Envs/ml/lib/python3.9/site-packages/torchvision/transforms/functional.py", line 1238, in gaussian_blur
    t_img = to_tensor(img)
  File "/home/kaustubh/Envs/ml/lib/python3.9/site-packages/torchvision/transforms/functional.py", line 129, in to_tensor
    np.array(pic, mode_to_nptype.get(pic.mode, np.uint8), copy=True)
TypeError: __array__() takes 1 positional argument but 2 were given


Here is the code:
def main():
    data_transforms = transforms.Compose([
        transforms.ToPILImage(),
        transforms.GaussianBlur(kernel_size=(11, 11), sigma=(0.1, 2.0)),
    ])

    im_dir = './test/'
    im_save_dir = './test/'

    for img in tqdm(sorted(os.listdir(im_dir))):
        ext = img.split('.')[1]
        im_name = img.split('.')[0]

        if ext == 'jpg':
            im_path = os.path.join(im_dir, img)

            image = cv2.imread(im_path)
            image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

            transformed_image = data_transforms(image)

if __name__ == "__main__":
    main()