Skip to content

Commit 23077c4

Browse files
committed
Fix up lint
1 parent 0ac1155 commit 23077c4

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

src/array_api_extra/_lib/_funcs.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,9 @@ def broadcast_shapes(*shapes: tuple[float | None, ...]) -> tuple[int | None, ...
281281
return tuple(out)
282282

283283

284-
def cov(m: Array, /, *, bias: bool = False, xp: ModuleType) -> Array: # numpydoc ignore=PR01,RT01
284+
def cov( # numpydoc ignore=PR01,RT01
285+
m: Array, /, *, bias: bool = False, xp: ModuleType
286+
) -> Array:
285287
"""See docstring in array_api_extra._delegation."""
286288
m = xp.asarray(m, copy=True)
287289
dtype = (

tests/test_funcs.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -627,9 +627,7 @@ def test_bias_batch(self, xp: ModuleType, bias: bool):
627627
n_var, n_obs = 3, 20
628628
m = rng.random((*batch_shape, n_var, n_obs))
629629
res = cov(xp.asarray(m), bias=bias)
630-
ref_list = [
631-
np.cov(m_, bias=bias) for m_ in np.reshape(m, (-1, n_var, n_obs))
632-
]
630+
ref_list = [np.cov(m_, bias=bias) for m_ in np.reshape(m, (-1, n_var, n_obs))]
633631
ref = np.reshape(np.stack(ref_list), (*batch_shape, n_var, n_var))
634632
xp_assert_close(res, xp.asarray(ref))
635633

0 commit comments

Comments
 (0)