make hack numpy 1.16 compatible

adrn · Jun 15, 2020 · b7ba1d0 · b7ba1d0
1 parent f008bc2
commit b7ba1d0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/thejoker/utils.py b/thejoker/utils.py
@@ -271,7 +271,10 @@ def wrapper(*args, **kwargs):
 def random_state_context(random_state):
     state = np.random.get_state()
     if random_state is not None:
-        np.random.seed(random_state.integers(2**32-1))  # HACK
+        try:
+            np.random.seed(random_state.integers(2**32-1))  # HACK
+        except AttributeError:
+            np.random.seed(random_state.randint(2**32-1))  # HACK
     try:
         yield
     finally: