diff --git a/hail/python/hail/linalg/blockmatrix.py b/hail/python/hail/linalg/blockmatrix.py index 3998b65452a..175e3dadff5 100644 --- a/hail/python/hail/linalg/blockmatrix.py +++ b/hail/python/hail/linalg/blockmatrix.py @@ -1303,14 +1303,7 @@ def T(self): if self.n_rows == 1 and self.n_cols == 1: return self - if self.n_rows == 1: - index_expr = [0] - elif self.n_cols == 1: - index_expr = [1] - else: - index_expr = [1, 0] - - return BlockMatrix(BlockMatrixBroadcast(self._bmir, index_expr, [self.n_cols, self.n_rows], self.block_size)) + return BlockMatrix(BlockMatrixBroadcast(self._bmir, [1, 0], [self.n_cols, self.n_rows], self.block_size)) def densify(self): """Restore all dropped blocks as explicit blocks of zeros.