From b7c1c4bca84031cdb115d38d30e86cded02f1fdd Mon Sep 17 00:00:00 2001
From: Gabriel Barth-Maron <gabrielbm@google.com>
Date: Mon, 24 May 2021 09:39:34 -0700
Subject: [PATCH] Expand scope for pylint: disable=not-supported-yet.

PiperOrigin-RevId: 375493980
Change-Id: Ia9188f01a255ca607bcf0d3d06a1de3ab3fd7a83
---
 acme/types.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/acme/types.py b/acme/types.py
index abfeebc072..7d841b97a5 100644
--- a/acme/types.py
+++ b/acme/types.py
@@ -23,11 +23,13 @@
 NestedArray = Any
 NestedTensor = Any
 
+# pytype: disable=not-supported-yet
 NestedSpec = Union[
     specs.Array,
     Iterable['NestedSpec'],
-    Mapping[Any, 'NestedSpec'],  # pytype: disable=not-supported-yet
+    Mapping[Any, 'NestedSpec'],
 ]
+# pytype: enable=not-supported-yet
 
 # TODO(b/144763593): Replace all instances of nest with the tensor/array types.
 Nest = Union[NestedArray, NestedTensor, NestedSpec]