Skip to content

Commit 7c3958f

Browse files
committed
fix B905
1 parent d3a5abf commit 7c3958f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

source/tests/pt/test_hybrid_muon.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,9 @@ def test_step(self) -> None:
104104
initial_params = [p.clone() for p in model.parameters()]
105105
optimizer.step()
106106

107-
for i, (p, init_p) in enumerate(zip(model.parameters(), initial_params)):
107+
for i, (p, init_p) in enumerate(
108+
zip(model.parameters(), initial_params, strict=True)
109+
):
108110
self.assertFalse(torch.allclose(p, init_p), f"Parameter {i} did not change")
109111

110112
def test_weight_decay(self) -> None:

0 commit comments

Comments
 (0)