Skip to content

GitLab

Unverified Commit 6af796ab authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

silence mypy (#6857)

parent cb60e97a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/prototype/transforms/_augment.py
View file @ 6af796ab
...@@ -125,7 +125,7 @@ class _BaseMixupCutmix(_RandomApplyTransform): ...@@ -125,7 +125,7 @@ class _BaseMixupCutmix(_RandomApplyTransform):
class RandomMixup(_BaseMixupCutmix): class RandomMixup(_BaseMixupCutmix):
def _get_params(self, flat_inputs: List[Any]) -> Dict[str, Any]: 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: def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
lam = params["lam"] lam = params["lam"]
...@@ -147,7 +147,7 @@ class RandomMixup(_BaseMixupCutmix): ...@@ -147,7 +147,7 @@ class RandomMixup(_BaseMixupCutmix):
class RandomCutmix(_BaseMixupCutmix): class RandomCutmix(_BaseMixupCutmix):
def _get_params(self, flat_inputs: List[Any]) -> Dict[str, Any]: 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) H, W = query_spatial_size(flat_inputs)
......
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