fixed local_index

lcgraham · May 11, 2016 · 40e0469 · 40e0469
1 parent 0eb3931
commit 40e0469
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bet/calculateP/calculateP.py b/bet/calculateP/calculateP.py
@@ -368,6 +368,7 @@ def estimate_volume(samples, lambda_emulate=None):
     #local_index = range(0+comm.rank, samples.shape[0], comm.size)
     local_index = np.array_split(np.arange(samples.shape[0]),
             comm.size)[comm.rank]
+    local_index = np.array(local_index, dtype='int64')
     lam_vol_local = lam_vol[local_index]
 
     return (lam_vol, lam_vol_local, local_index)