Merge pull request #45 from cclauss/patch-1

xrange() was removed for Python on 1/1/2020
google-deepmind · Jul 23, 2021 · f5981c6 · f5981c6
2 parents d3d94b3 + 1e85949
commit f5981c6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/dnc/addressing_test.py b/dnc/addressing_test.py
@@ -25,6 +25,11 @@
 from dnc import addressing
 from dnc import util
 
+try:
+  xrange
+except NameError:
+  xrange = range
+
 
 class WeightedSoftmaxTest(tf.test.TestCase):
 

diff --git a/dnc/util_test.py b/dnc/util_test.py
@@ -23,6 +23,11 @@
 
 from dnc import util
 
+try:
+  xrange
+except NameError:
+  xrange = range
+
 
 class BatchInvertPermutation(tf.test.TestCase):