Update: test

heptaflar · heptaflar · commit 0df9ebe6c201 · 2026-04-24T16:48:14.000+02:00
diff --git a/test/oceancolour/test_qaa.py b/test/oceancolour/test_qaa.py
@@ -128,10 +128,10 @@ def diagonalize(U: np.ndarray) -> np.ndarray:  # noqa: N806
     """
     return np.stack(
         [
-            np.diag(
-                U[i].reshape((-1, np.prod(U.shape[1 + U.ndim // 2 :])))
-            ).reshape(U.shape[1 + U.ndim // 2 :])
-            for i in range(U.shape[0])
+            np.diag(_.reshape((np.prod(_.shape[_.ndim // 2 :]), -1))).reshape(
+                _.shape[_.ndim // 2 :]
+            )
+            for _ in U
         ]
     )
 

Original file line number	Diff line number	Diff line change
`@@ -128,10 +128,10 @@ def diagonalize(U: np.ndarray) -> np.ndarray: # noqa: N806`
`128`	`128`	`"""`
`129`	`129`	`return np.stack(`
`130`	`130`	`[`
`131`		`- np.diag(`
`132`		`- U[i].reshape((-1, np.prod(U.shape[1 + U.ndim // 2 :])))`
`133`		`- ).reshape(U.shape[1 + U.ndim // 2 :])`
`134`		`- for i in range(U.shape[0])`
	`131`	`+ np.diag(_.reshape((np.prod(_.shape[_.ndim // 2 :]), -1))).reshape(`
	`132`	`+ _.shape[_.ndim // 2 :]`
	`133`	`+ )`
	`134`	`+ for _ in U`
`135`	`135`	`]`
`136`	`136`	`)`
`137`	`137`