Replace split_by_diag_blocks() by SymPy's stock get_diag_blocks().