random.utilities: Run guvectorize only when necessary

rht · rht · commit 673131ebc69a · 2018-12-11T04:36:15.000Z
diff --git a/quantecon/random/utilities.py b/quantecon/random/utilities.py
@@ -56,8 +56,14 @@ def probvec(m, k, random_state=None, parallel=True):
 
     # Parse Parallel Option #
     if parallel:
+        _probvec_parallel = guvectorize(
+            ['(f8[:], f8[:])'], '(n), (k)', nopython=True, target='parallel'
+        )(_probvec)
         _probvec_parallel(r, x)
     else:
+        _probvec_cpu = guvectorize(
+            ['(f8[:], f8[:])'], '(n), (k)', nopython=True, target='cpu'
+        )(_probvec)
         _probvec_cpu(r, x)
 
     return x
@@ -88,13 +94,6 @@ def _probvec(r, out):
         out[i] = r[i] - r[i-1]
     out[n] = 1 - r[n-1]
 
-_probvec_parallel = guvectorize(
-    ['(f8[:], f8[:])'], '(n), (k)', nopython=True, target='parallel'
-    )(_probvec)
-_probvec_cpu = guvectorize(
-    ['(f8[:], f8[:])'], '(n), (k)', nopython=True, target='cpu'
-    )(_probvec)
-
 
 @jit
 def sample_without_replacement(n, k, num_trials=None, random_state=None):
