Update src/array_api_extra/_delegation.py

bruAristimunha · betatim · web-flow · commit 83b7e1becd85 · 2026-04-20T11:54:06.000+02:00
Co-authored-by: Tim Head &lt;betatim@gmail.com&gt;
diff --git a/src/array_api_extra/_delegation.py b/src/array_api_extra/_delegation.py
@@ -237,7 +237,7 @@ def cov(
                 else xp.asarray(frequency_weights, device=device)
             )
             aw = None if weights is None else xp.asarray(weights, device=device)
-            return xp.cov(m, correction=int(correction), fweights=fw, aweights=aw)
+            return xp.cov(m, correction=correction, fweights=fw, aweights=aw)
         # `dask.array.cov` forces `.compute()` whenever weights are given:
         # its internal `if fact <= 0` check on a lazy 0-D scalar triggers
         # materialization. Route to the generic impl, which is fully lazy

Original file line number	Diff line number	Diff line change
`@@ -237,7 +237,7 @@ def cov(`
`237`	`237`	`else xp.asarray(frequency_weights, device=device)`
`238`	`238`	`)`
`239`	`239`	`aw = None if weights is None else xp.asarray(weights, device=device)`
`240`		`- return xp.cov(m, correction=int(correction), fweights=fw, aweights=aw)`
	`240`	`+ return xp.cov(m, correction=correction, fweights=fw, aweights=aw)`
`241`	`241`	# `dask.array.cov` forces `.compute()` whenever weights are given:
`242`	`242`	# its internal `if fact <= 0` check on a lazy 0-D scalar triggers
`243`	`243`	`# materialization. Route to the generic impl, which is fully lazy`