Skip to content

Commit

Permalink
fix(granite): apply ruff recommendation
Browse files Browse the repository at this point in the history
  • Loading branch information
dacorvo committed Dec 24, 2024
1 parent c2c6557 commit 93e7f31
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/neuron/models/granite/hlo.py
Original file line number Diff line number Diff line change
Expand Up @@ -325,7 +325,7 @@ def layer(
)
if self.neuron_config.fuse_mlp:
assert all(
map(lambda x: not (x), [in0_weight, in1_weight, out_weight, in0_scales, in1_scales, out_scales])
(not (x) for x in [in0_weight, in1_weight, out_weight, in0_scales, in1_scales, out_scales])
), "in0, in1 and out weights have to be None"
in0_weight, in0_scales = mlp_in_weight, mlp_in_scales
out_weight, out_scales = mlp_out_weight, mlp_out_scales
Expand Down

0 comments on commit 93e7f31

Please sign in to comment.