Expand scope for pylint: disable=not-supported-yet.

PiperOrigin-RevId: 375493980 Change-Id: Ia9188f01a255ca607bcf0d3d06a1de3ab3fd7a83
google-deepmind · May 24, 2021 · b7c1c4b · b7c1c4b
1 parent 46877a9
commit b7c1c4b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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]