Make this code compatible with Python 3.10.

PiperOrigin-RevId: 473345210
google · Sep 9, 2022 · 050e0b6 · 050e0b6
1 parent 78d2b6e
commit 050e0b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trax/tf_numpy/numpy_impl/tests/arrays_test.py b/trax/tf_numpy/numpy_impl/tests/arrays_test.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 
 """Tests for ndarray."""
-import collections
+from collections import abc
 
 import numpy as np
 import tensorflow.compat.v2 as tf
@@ -170,7 +170,7 @@ def testBool(self):
 
   def testHash(self):
     a = t2a(tf.convert_to_tensor(value=10))
-    self.assertNotIsInstance(a, collections.Hashable)
+    self.assertNotIsInstance(a, abc.Hashable)
     with self.assertRaisesWithPredicateMatch(
         TypeError, r'unhashable type'):
       hash(a)