Skip to content

Commit 54c3799

Browse files
committed
Format code with ruff
1 parent 6fff0e6 commit 54c3799

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

analysis/generate_figures.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,7 @@ def accuracy_comparison_plot(df: pd.DataFrame, output_path: Path, augment: str =
2525

2626
for ax, dataset in zip(axes, datasets):
2727
data = df[df["dataset"] == dataset]
28-
pivot = data.pivot_table(
29-
values="best_acc", index="model", columns="version", aggfunc="mean"
30-
)
28+
pivot = data.pivot_table(values="best_acc", index="model", columns="version", aggfunc="mean")
3129

3230
pivot.plot(kind="bar", ax=ax, rot=45)
3331
ax.set_title(dataset.upper())

bitnet/nn/bitconv2d.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,5 @@ def forward(self, x: Tensor) -> Tensor:
1818
w_quant = ste_ternary(self.weight)
1919
beta = self.weight.abs().mean()
2020

21-
out = f.conv2d(
22-
x_quant, w_quant, self.bias, self.stride, self.padding, self.dilation, self.groups
23-
)
21+
out = f.conv2d(x_quant, w_quant, self.bias, self.stride, self.padding, self.dilation, self.groups)
2422
return dequantize(out, gamma, beta, self.num_bits)

0 commit comments

Comments
 (0)