From 20dd118f2b98ff03a28c24d1cb99bf0d79e3d5c0 Mon Sep 17 00:00:00 2001
From: Iurii Kemaev <iukemaev@google.com>
Date: Fri, 20 Oct 2023 04:15:59 -0700
Subject: [PATCH] Silence `invalid-name` warning.

PiperOrigin-RevId: 575177733
---
 chex/_src/pytypes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/chex/_src/pytypes.py b/chex/_src/pytypes.py
index 0f70d15..1fa4935 100644
--- 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  # pytype:disable=invalid-name
 else:
-  ArrayDType = Any
+  ArrayDType = Any  # pytype:disable=invalid-name