Type fix in transformers.py (#6376)

Update `RandomPhotometricDistort` `__init__` argument to correct types.

Type fix in transformers.py (#6376)
Update `RandomPhotometricDistort` `__init__` argument to correct types.
96dbada4 · vcwai · GitHub · 96aa3d92 · 96dbada4
Unverified Commit 96dbada4 authored Aug 08, 2022 by vcwai Committed by GitHub Aug 08, 2022
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

references/detection/transforms.py references/detection/transforms.py +4 -4

No files found.
--- a/references/detection/transforms.py
+++ b/references/detection/transforms.py
@@ -218,10 +218,10 @@ class RandomZoomOut(nn.Module):
 class RandomPhotometricDistort(nn.Module):
    def __init__(
        self,
-        contrast: Tuple[float] = (0.5, 1.5),
-        saturation: Tuple[float] = (0.5, 1.5),
-        hue: Tuple[float] = (-0.05, 0.05),
-        brightness: Tuple[float] = (0.875, 1.125),
+        contrast: Tuple[float, float] = (0.5, 1.5),
+        saturation: Tuple[float, float] = (0.5, 1.5),
+        hue: Tuple[float, float] = (-0.05, 0.05),
+        brightness: Tuple[float, float] = (0.875, 1.125),
        p: float = 0.5,
    ):
        super().__init__()