Skip to content

Commit 92e5638

Browse files
Apply ruff-format to the rank-4 RMSNorm test
1 parent eff0eac commit 92e5638

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

src/mobius/rewrite_rules/_htp_rank4_rmsnorm_test.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,10 @@ def _rmsnorm_model(shape: list[int]) -> ir.Model:
2020
rng = np.random.default_rng(0)
2121
x = ir.Value(name="x", shape=ir.Shape(shape), type=ir.TensorType(ir.DataType.FLOAT))
2222
graph = ir.Graph(
23-
inputs=[x], outputs=[], nodes=[], name="rmsnorm",
23+
inputs=[x],
24+
outputs=[],
25+
nodes=[],
26+
name="rmsnorm",
2427
opset_imports={"": OPSET_VERSION},
2528
)
2629
op = GraphBuilder(graph).op

0 commit comments

Comments
 (0)