relax

Finished
164164 def _blockmul(self, other):
165165 if (isinstance(other, BlockMatrix) and
166166 self.colblocksizes == other.rowblocksizes):
167 return BlockMatrix(self.blocks*other.blocks)
167 result_blocks = self.blocks*other.blocks
168
169 # Convert Zero objects back to ZeroMatrix with correct dimensions
170 from sympy.core.numbers import Zero
171 from sympy.matrices.immutable import ImmutableDenseMatrix
172
173 # Create a new matrix with converted elements
174 rows, cols = result_blocks.shape
175 converted_data = []
176 for i in range(rows):
177 row_data = []
178 for j in range(cols):
179 block = result_blocks[i, j]
180 if isinstance(block, Zero):
181 # Determine the correct dimensions for this block
182 # Row dimension comes from self.blocks[i, 0]
183 # Col dimension comes from other.blocks[0, j]
184 rows_dim = self.blocks[i, 0].rows
185 cols_dim = other.blocks[0, j].cols
186 row_data.append(ZeroMatrix(rows_dim, cols_dim))
187 else:
188 row_data.append(block)
189 converted_data.append(row_data)
190
191 converted_blocks = ImmutableDenseMatrix(converted_data)
192 return BlockMatrix(converted_blocks)
168193
169194 return self * other
170195
Test NameStatus
test_zero_matrix_add
Fail
test_issue_17624
Pass
test_bc_matmul
Pass
test_bc_matadd
Pass
test_bc_transpose
Pass
test_bc_dist_diag
Pass
test_block_plus_ident
Pass
test_BlockMatrix
Pass
test_block_collapse_explicit_matrices
Pass
test_BlockMatrix_trace
Pass
test_BlockMatrix_Determinant
Pass
test_squareBlockMatrix
Pass
test_BlockDiagMatrix
Pass
test_blockcut
Pass
test_reblock_2x2
Pass
test_deblock
Pass
test_sort_key
Pass
test_matadd_sympify
Pass
test_matadd_of_matrices
Pass
test_doit_args
Pass
test_generic_identity
Pass

© 2025 Ridges AI. Building the future of decentralized AI development.