Handle rowvar and y properly

antonwolfy · antonwolfy · commit 852f7a34323d · 2025-02-11T20:39:23.000+01:00
diff --git a/dpnp/dpnp_utils/dpnp_utils_statistics.py b/dpnp/dpnp_utils/dpnp_utils_statistics.py
@@ -139,7 +139,10 @@ def dpnp_cov(
         )
 
     if y is not None:
+        y_ndim = y.ndim
         y = dpnp.array(y, copy=None, ndmin=2, dtype=dtype)
+        if not rowvar and y_ndim != 1:
+            y = y.T
         x = dpnp.concatenate((x, y), axis=0)
 
     # get the product of frequencies and weights

Original file line number	Diff line number	Diff line change
`@@ -139,7 +139,10 @@ def dpnp_cov(`
`139`	`139`	`)`
`140`	`140`
`141`	`141`	`if y is not None:`
	`142`	`+ y_ndim = y.ndim`
`142`	`143`	`y = dpnp.array(y, copy=None, ndmin=2, dtype=dtype)`
	`144`	`+ if not rowvar and y_ndim != 1:`
	`145`	`+ y = y.T`
`143`	`146`	`x = dpnp.concatenate((x, y), axis=0)`
`144`	`147`
`145`	`148`	`# get the product of frequencies and weights`