Fix typo in count_nonzero caught in review

ndgrigorian · ndgrigorian · commit 6c8a61f3034c · 2024-07-31T13:01:01.000-07:00
diff --git a/dpctl/tensor/_reduction.py b/dpctl/tensor/_reduction.py
@@ -814,5 +814,5 @@ def count_nonzero(x, /, *, axis=None, keepdims=False, out=None):
         axis=axis,
         dtype=ti.default_device_index_type(x.sycl_device),
         keepdims=keepdims,
-        out=None,
+        out=out,
     )

Original file line number	Diff line number	Diff line change
`@@ -814,5 +814,5 @@ def count_nonzero(x, /, *, axis=None, keepdims=False, out=None):`
`814`	`814`	`axis=axis,`
`815`	`815`	`dtype=ti.default_device_index_type(x.sycl_device),`
`816`	`816`	`keepdims=keepdims,`
`817`		`- out=None,`
	`817`	`+ out=out,`
`818`	`818`	`)`