Back to size

paddyroddy · paddyroddy · commit df04e6aa0bd0 · 2026-01-29T18:14:49.000Z
diff --git a/glass/galaxies.py b/glass/galaxies.py
@@ -152,7 +152,7 @@ def redshifts_from_nz(
         )
         total += count_out[k]
 
-    assert total == redshifts.shape[0]  # noqa: S101
+    assert total == redshifts.size  # noqa: S101
 
     return redshifts  # ty: ignore[invalid-return-type]
 

Original file line number	Diff line number	Diff line change
`@@ -152,7 +152,7 @@ def redshifts_from_nz(`
`152`	`152`	`)`
`153`	`153`	`total += count_out[k]`
`154`	`154`
`155`		`- assert total == redshifts.shape[0] # noqa: S101`
	`155`	`+ assert total == redshifts.size # noqa: S101`
`156`	`156`
`157`	`157`	`return redshifts # ty: ignore[invalid-return-type]`
`158`	`158`