Minor corrections

Author: datumbox

torchvision/prototype/transforms/_augment.py

@@ -55,7 +55,7 @@ def _get_params(self, sample: Any) -> Dict[str, Any]:
 
         if value is not None and not (len(value) in (1, img_c)):
             raise ValueError(
-                "If value is a sequence, it should have either a single value or " f"{img_c} (number of input channels)"
+                f"If value is a sequence, it should have either a single value or {img_c} (number of input channels)"
             )
 
         area = img_h * img_w

torchvision/prototype/transforms/_geometry.py

@@ -34,7 +34,7 @@ def __init__(
         interpolation: InterpolationMode = InterpolationMode.BILINEAR,
     ) -> None:
         super().__init__()
-        self.size = [size, size] if isinstance(size, int) else list(size)
+        self.size = [size] if isinstance(size, int) else list(size)
         self.interpolation = interpolation
 
     def _transform(self, input: Any, params: Dict[str, Any]) -> Any: