fixed local index

lcgraham · May 11, 2016 · 0eb3931 · 0eb3931
1 parent b6efc02
commit 0eb3931
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/bet/calculateP/calculateP.py b/bet/calculateP/calculateP.py
@@ -432,6 +432,7 @@ def exact_volume_1D(samples, input_domain, distribution='uniform', a=None,
     # Set up local arrays for parallelism
     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)