Disable validate_args

rlworkgroup · Mar 9, 2021 · 73dd20e · 73dd20e
1 parent 62bbc5c
commit 73dd20e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/garage/torch/distributions/tanh_normal.py b/src/garage/torch/distributions/tanh_normal.py
@@ -19,7 +19,8 @@ class TanhNormal(torch.distributions.Distribution):
     """ # noqa: 501
 
     def __init__(self, loc, scale):
-        self._normal = Independent(Normal(loc, scale), 1)
+        # pylint: disable=W0223
+        self._normal = Independent(Normal(loc, scale), 1, validate_args=False)
         super().__init__()
 
     def log_prob(self, value, pre_tanh_value=None, epsilon=1e-6):