Delete stale "pylint: disable".

PiperOrigin-RevId: 424641627
google-deepmind · Jan 27, 2022 · bce1753 · bce1753
1 parent 41cd667
commit bce1753
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distrax/_src/distributions/gumbel.py b/distrax/_src/distributions/gumbel.py
@@ -95,7 +95,7 @@ def _sample_n_and_log_prob(self, key: PRNGKey, n: int) -> Tuple[Array, Array]:
     log_prob = -(rnd + jnp.exp(-rnd)) - jnp.log(self._scale)
     return samples, log_prob
 
-  def entropy(self) -> Array:  # pylint:disable=arguments-differ
+  def entropy(self) -> Array:
     """Calculates the Shannon entropy (in nats)."""
     return jnp.log(self._scale) + 1. + jnp.euler_gamma