Fix sampling bug

wesselb · Aug 28, 2021 · 531dedd · 531dedd
1 parent 2bb0321
commit 531dedd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/matrix/ops/sample.py b/matrix/ops/sample.py
@@ -27,7 +27,9 @@ def sample(state: B.RandomState, a, num: B.Int = 1):
 
 @B.dispatch
 def sample(a, num: B.Int = 1):
-    return sample(B.global_random_state(a), a, num=num)[1]
+    state, res = sample(B.global_random_state(a), a, num=num)
+    B.set_global_random_state(state)
+    return res
 
 
 @B.dispatch

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 requirements = [
     "numpy>=1.16",
     "plum-dispatch>=1.5.3",
-    "backends>=1.4.1",
+    "backends>=1.4.3",
     "wbml>=0.3",
     "algebra",
 ]