Add missing type

datumbox · datumbox · commit bd2dc1704355 · 2021-09-02T16:30:14.000+01:00
diff --git a/torchvision/transforms/autoaugment.py b/torchvision/transforms/autoaugment.py
@@ -369,7 +369,7 @@ def _augmentation_space(self, num_bins: int) -> Dict[str, Tuple[Tensor, bool]]:
             "Invert": (torch.tensor(0.0), False),
         }
 
-    def forward(self, img: Tensor):
+    def forward(self, img: Tensor) -> Tensor:
         """
             img (PIL Image or Tensor): Image to be transformed.
 

Original file line number	Diff line number	Diff line change
`@@ -369,7 +369,7 @@ def _augmentation_space(self, num_bins: int) -> Dict[str, Tuple[Tensor, bool]]:`
`369`	`369`	`"Invert": (torch.tensor(0.0), False),`
`370`	`370`	`}`
`371`	`371`
`372`		`- def forward(self, img: Tensor):`
	`372`	`+ def forward(self, img: Tensor) -> Tensor:`
`373`	`373`	`"""`
`374`	`374`	`img (PIL Image or Tensor): Image to be transformed.`
`375`	`375`