Skip to content

Commit

Permalink
minor revision
Browse files Browse the repository at this point in the history
  • Loading branch information
ajpotts committed Feb 5, 2025
1 parent 2c04b0a commit a42a54b
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/array_api/binary_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,12 +19,13 @@ class TestArrayCreation:
def test_binops(self, op, dtype):

for shape_a, shape_b in zip(SHAPE_A, SHAPE_B):
x = xp.asarray(ak.randint(0, 100, shape_a, dtype=dtype, seed=SEED)) # noqa F841
x = xp.asarray(
ak.randint(0, 100, shape_a, dtype=dtype, seed=SEED)
) # noqa F841
y = xp.asarray(ak.randint(0, 100, shape_b, dtype=dtype, seed=SEED))

z = eval("x " + op + " y")
ybc = xp.broadcast_to(y, z.shape) # noqa F841
xp.broadcast_to(y, z.shape)

if z.ndim == 1:
for i in range(shape_a[0]):
Expand Down

0 comments on commit a42a54b

Please sign in to comment.