Unverified Commit 01b0a004 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

remove magic comma in some places (#5880)

parent 66ed6937
......@@ -1030,25 +1030,9 @@ def test_resized_crop(device, mode):
(F_t.adjust_contrast, (1.0,)),
(F_t.adjust_hue, (-0.5,)),
(F_t.adjust_saturation, (2.0,)),
(
F_t.pad,
(
[
2,
],
2,
"constant",
),
),
(F_t.pad, ([2], 2, "constant")),
(F_t.resize, ([10, 11],)),
(
F_t.perspective,
(
[
0.2,
]
),
),
(F_t.perspective, ([0.2])),
(F_t.gaussian_blur, ((2, 2), (0.7, 0.5))),
(F_t.invert, ()),
(F_t.posterize, (0,)),
......
......@@ -570,12 +570,7 @@ def _cast_squeeze_out(img: Tensor, need_cast: bool, need_squeeze: bool, out_dtyp
def _apply_grid_transform(img: Tensor, grid: Tensor, mode: str, fill: Optional[List[float]]) -> Tensor:
img, need_cast, need_squeeze, out_dtype = _cast_squeeze_in(
img,
[
grid.dtype,
],
)
img, need_cast, need_squeeze, out_dtype = _cast_squeeze_in(img, [grid.dtype])
if img.shape[0] > 1:
# Apply same grid to a batch of images
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment