Skip to content

Commit 0030660

Browse files
committed
Apply ruff format
1 parent af8ef07 commit 0030660

1 file changed

Lines changed: 18 additions & 3 deletions

File tree

tests/ops/test_ops.py

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2924,12 +2924,27 @@ def forward(self, x: Tensor) -> Tensor:
29242924
# mean.default: global mean (dim=None), int32 input → float32 output.
29252925
# torch promotes the operand to the requested dtype BEFORE averaging.
29262926
(torch.randint(0, 10, (3, 4), dtype=torch.int32), None, False, torch.float32),
2927-
(torch.randint(-5, 5, (2, 3, 4), dtype=torch.int32), None, False, torch.float32),
2927+
(
2928+
torch.randint(-5, 5, (2, 3, 4), dtype=torch.int32),
2929+
None,
2930+
False,
2931+
torch.float32,
2932+
),
29282933
# mean.dim: reduce along specific dimensions, int32 input → float32.
29292934
(torch.randint(0, 10, (3, 4), dtype=torch.int32), 1, False, torch.float32),
29302935
(torch.randint(0, 10, (3, 4), dtype=torch.int32), 0, True, torch.float32),
2931-
(torch.randint(-5, 5, (2, 3, 4), dtype=torch.int32), [0, 2], False, torch.float32),
2932-
(torch.randint(-5, 5, (2, 3, 4), dtype=torch.int32), [0, 2], True, torch.float32),
2936+
(
2937+
torch.randint(-5, 5, (2, 3, 4), dtype=torch.int32),
2938+
[0, 2],
2939+
False,
2940+
torch.float32,
2941+
),
2942+
(
2943+
torch.randint(-5, 5, (2, 3, 4), dtype=torch.int32),
2944+
[0, 2],
2945+
True,
2946+
torch.float32,
2947+
),
29332948
# dtype-agnostic: root cause is independent of the specific int operand
29342949
# dtype or the specific float target dtype.
29352950
(torch.randint(0, 10, (3, 4), dtype=torch.int64), 1, False, torch.float32),

0 commit comments

Comments
 (0)