Skip to content

Commit 0018e4a

Browse files
TimDettmersclaude
andcommitted
style: Fix ruff format violation in test_linear4bit.py
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 65c9fff commit 0018e4a

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/test_linear4bit.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -276,9 +276,7 @@ def test_quant_storage_shard_roundtrip(device, quant_type, quant_storage):
276276
reassembled = torch.cat(shards).reshape(qB.shape)
277277

278278
assert reassembled.dtype == qB.dtype
279-
assert torch.equal(
280-
reassembled.view(torch.uint8), qB.view(torch.uint8)
281-
), "Bytes changed after shard roundtrip"
279+
assert torch.equal(reassembled.view(torch.uint8), qB.view(torch.uint8)), "Bytes changed after shard roundtrip"
282280

283281
out = bnb.functional.gemv_4bit(A, reassembled.t(), state=state)
284282
torch.testing.assert_close(out, ref)

0 commit comments

Comments
 (0)