Silence invalid-name warning.

PiperOrigin-RevId: 575225107
google-deepmind · Oct 20, 2023 · 3c9af1b · 3c9af1b
1 parent 626d697
commit 3c9af1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chex/_src/pytypes.py b/chex/_src/pytypes.py
@@ -60,6 +60,6 @@
 # TODO(iukemaev, jakevdp): upgrade minimum jax version & remove this condition.
 if hasattr(jax.typing, 'DTypeLike'):
   # jax version 0.4.19 or newer
-  ArrayDType = jax.typing.DTypeLike
+  ArrayDType = jax.typing.DTypeLike  # pylint:disable=invalid-name
 else:
-  ArrayDType = Any
+  ArrayDType = Any  # pylint:disable=invalid-name