Skip to content

Commit

Permalink
style: pre-commit fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
pre-commit-ci[bot] committed Dec 31, 2023
1 parent ee891a9 commit bcf9bcf
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions tests/myarray.py
Original file line number Diff line number Diff line change
Expand Up @@ -257,11 +257,14 @@ def _complex_p(x: MyArray, y: MyArray) -> MyArray:

@register(lax.concatenate_p)
def _concatenate_p(
operand0: MyArray, *operands: MyArray | DenseArrayValue, dimension: Any,
operand0: MyArray,
*operands: MyArray | DenseArrayValue,
dimension: Any,
) -> MyArray:
return MyArray(
lax.concatenate(
[operand0.array] + [op.array for op in operands], dimension=dimension,
[operand0.array] + [op.array for op in operands],
dimension=dimension,
),
)

Expand Down

0 comments on commit bcf9bcf

Please sign in to comment.