Fixed bad merge

Author: vfdev-5

test/test_prototype_transforms_functional.py

@@ -228,7 +228,7 @@ def resize_bounding_box():
 @register_kernel_info_from_sample_inputs_fn
 def affine_image_tensor():
     for image, angle, translate, scale, shear in itertools.product(
-        make_segmentation_masks(extra_dims=((), (4,))),
+        make_images(extra_dims=((), (4,))),
         [-87, 15, 90],  # angle
         [5, -5],  # translate
         [0.77, 1.27],  # scale
@@ -267,7 +267,7 @@ def affine_bounding_box():
 @register_kernel_info_from_sample_inputs_fn
 def affine_segmentation_mask():
     for image, angle, translate, scale, shear in itertools.product(
-        make_images(extra_dims=()),
+        make_segmentation_masks(extra_dims=((), (4,))),
         [-87, 15, 90],  # angle
         [5, -5],  # translate
         [0.77, 1.27],  # scale
@@ -279,6 +279,7 @@ def affine_segmentation_mask():
             translate=(translate, translate),
             scale=scale,
             shear=(shear, shear),
+        )
 
 
 @register_kernel_info_from_sample_inputs_fn