Replace from numpy.random import poisson with import numpy as np …

…in `cirq-core/cirq/contrib/acquaintance/gates_test.py` (#4451) Fixes: #4448
quantumlib · Aug 20, 2021 · 662de03 · 662de03
1 parent 11f751f
commit 662de03
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cirq-core/cirq/contrib/acquaintance/gates_test.py b/cirq-core/cirq/contrib/acquaintance/gates_test.py
@@ -17,7 +17,7 @@
 from string import ascii_lowercase as alphabet
 from typing import Optional, Sequence, Tuple
 
-from numpy.random import poisson
+import numpy as np
 import pytest
 
 import cirq
@@ -233,7 +233,7 @@ def test_swap_network_init_error():
 
 
 part_lens_and_acquaintance_sizes = [
-    [[l + 1 for l in poisson(size=n_parts, lam=lam)], poisson(4)]
+    [[l + 1 for l in np.random.poisson(size=n_parts, lam=lam)], np.random.poisson(4)]
     for n_parts, lam in product(range(2, 20, 3), range(1, 4))
 ]