Fix index table base calculation for MR x MR

Crunch-io · Oct 17, 2018 · 5f73986 · 5f73986
1 parent d8f86d6
commit 5f73986
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/cr/cube/cube_slice.py b/src/cr/cube/cube_slice.py
@@ -355,10 +355,10 @@ def _prepare_index_base(self, axis):
             base = base / np.sum(base, axis=1)[:, None]
             return base[:, 0]
         elif isinstance(self.mr_dim_ind, tuple) and axis in self.mr_dim_ind:
-            total = np.sum(base, axis=(axis + 1))[0]
+            total = np.sum(base, axis=(axis + 1))
             if axis == 0:
-                return base[:, 0] / total[0]
-            return base[0, :, 0] / total
+                return base[:, 0, 0] / total[:, 0]
+            return base[0, :, 0] / total[0]
 
         if axis == 0 and self.mr_dim_ind is not None:
             base = base[:, 0]