silence mypy (#6857)

6af796ab · Philip Meier · GitHub · cb60e97a · 6af796ab
Unverified Commit 6af796ab authored Oct 28, 2022 by Philip Meier Committed by GitHub Oct 28, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchvision/prototype/transforms/_augment.py torchvision/prototype/transforms/_augment.py +2 -2

No files found.
--- a/torchvision/prototype/transforms/_augment.py
+++ b/torchvision/prototype/transforms/_augment.py
@@ -125,7 +125,7 @@ class _BaseMixupCutmix(_RandomApplyTransform):

 class RandomMixup(_BaseMixupCutmix):
    def _get_params(self, flat_inputs: List[Any]) -> Dict[str, Any]:
-        return dict(lam=float(self._dist.sample(())))
+        return dict(lam=float(self._dist.sample(())))  # type: ignore[arg-type]

    def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
        lam = params["lam"]
@@ -147,7 +147,7 @@ class RandomMixup(_BaseMixupCutmix):

 class RandomCutmix(_BaseMixupCutmix):
    def _get_params(self, flat_inputs: List[Any]) -> Dict[str, Any]:
-        lam = float(self._dist.sample(()))
+        lam = float(self._dist.sample(()))  # type: ignore[arg-type]

        H, W = query_spatial_size(flat_inputs)