Skip to content

Commit 5584b06

Browse files
committed
Fix scaled_matrices_2_plus_rows
- It should not contain zero matrices
1 parent e456c48 commit 5584b06

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

tests/unit/aggregation/_inputs.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,9 +106,7 @@ def _generate_semi_orthonormal_complement(Q: Tensor) -> Tensor:
106106
scaled_matrices = [scale * matrix for scale in _scales for matrix in matrices]
107107
zero_matrices = [torch.zeros([m, n]) for m, n in _zero_matrices_shapes]
108108
matrices_2_plus_rows = [matrix for matrix in matrices + zero_matrices if matrix.shape[0] >= 2]
109-
scaled_matrices_2_plus_rows = [
110-
matrix for matrix in scaled_matrices + zero_matrices if matrix.shape[0] >= 2
111-
]
109+
scaled_matrices_2_plus_rows = [matrix for matrix in scaled_matrices if matrix.shape[0] >= 2]
112110
strong_stationary_matrices = [
113111
_generate_strong_stationary_matrix(m, n) for m, n in _stationary_matrices_shapes
114112
]

0 commit comments

Comments
 (0)