Add typical_matrices_2_plus_rows

ValerianRey · ValerianRey · commit 98897dd79e25 · 2025-03-22T13:57:47.000+01:00
diff --git a/tests/unit/aggregation/_inputs.py b/tests/unit/aggregation/_inputs.py
@@ -115,3 +115,4 @@ def _generate_semi_orthonormal_complement(Q: Tensor) -> Tensor:
 
 matrices_2_plus_rows = [matrix for matrix in matrices + zero_matrices if matrix.shape[0] >= 2]
 scaled_matrices_2_plus_rows = [matrix for matrix in scaled_matrices if matrix.shape[0] >= 2]
+typical_matrices_2_plus_rows = [matrix for matrix in typical_matrices if matrix.shape[0] >= 2]

Original file line number	Diff line number	Diff line change
`@@ -115,3 +115,4 @@ def _generate_semi_orthonormal_complement(Q: Tensor) -> Tensor:`
`115`	`115`
`116`	`116`	`matrices_2_plus_rows = [matrix for matrix in matrices + zero_matrices if matrix.shape[0] >= 2]`
`117`	`117`	`scaled_matrices_2_plus_rows = [matrix for matrix in scaled_matrices if matrix.shape[0] >= 2]`
	`118`	`+typical_matrices_2_plus_rows = [matrix for matrix in typical_matrices if matrix.shape[0] >= 2]`