Fix shape_test which complains on non-integer slice.

Change: 147399432

Author: Joshua V. Dillon

tensorflow/contrib/distributions/python/kernel_tests/shape_test.py

@@ -50,7 +50,7 @@ def _random_sample(self, sample_shape, dtype=np.float32):
  return self._rng.random_sample(sample_shape).astype(dtype)
 
  def _get_expected(self, x, batch_ndims, event_ndims, expand_batch_dim):
- n = x.ndim - batch_ndims - event_ndims
+ n = int(x.ndim - batch_ndims - event_ndims)
  sample_shape = x.shape[0:n]
  y = np.reshape(x, np.concatenate([[-1], x.shape[n:]], 0))
  y = np.transpose(y, np.roll(np.arange(y.ndim), -1))