Skip to content

Commit 2d51e63

Browse files
authored
Fix linter (#7025)
Test plan: CI
1 parent 6bf9838 commit 2d51e63

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

examples/models/llama/source_transformation/apply_spin_quant_r1_r2.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,8 @@ def fuse_ln_linear(
148148
linear.bias.data = linear.bias.data.to(dtype=torch.float32) + torch.matmul(
149149
# pyre-fixme[6]: For 2nd argument expected `Tensor` but got
150150
# `Union[Tensor, Module]`.
151-
W_, layernorm.bias.to(dtype=torch.float32)
151+
W_,
152+
layernorm.bias.to(dtype=torch.float32),
152153
)
153154
linear.bias.data = linear.bias.data.to(linear_dtype)
154155

exir/emit/_emitter.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1635,7 +1635,8 @@ def plan(self) -> ExecutionPlan:
16351635
# empty list.
16361636
non_const_buffer_sizes=typing.cast(
16371637
# pyre-fixme[29]: `Union[BoundMethod[typing.Callable(torch._C.TensorB...
1638-
List[int], self.module.meta["non_const_buffer_sizes"]
1638+
List[int],
1639+
self.module.meta["non_const_buffer_sizes"],
16391640
),
16401641
container_meta_type=self.container_meta_type,
16411642
)

0 commit comments

Comments
 (0)