use integer division for numpy future-proofing

dit · May 10, 2016 · 03755b2 · 03755b2
1 parent 4501a11
commit 03755b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dit/abstractdist.py b/dit/abstractdist.py
@@ -321,7 +321,7 @@ def brute_marginal_array(d, rvs, rv_mode=None):
     # Apply nonzero, use [1] to get only the columns
     nz = np.nonzero(arr)[1]
     n_rows = len(marginal._sample_space)
-    arr = nz.reshape((n_rows, len(nz) / n_rows))
+    arr = nz.reshape((n_rows, len(nz) // n_rows))
 
     return arr