fix(test): add strict parameter to zip() calls in test_adamuon.py (#221)

* Initial plan

* fix(test): add strict=True parameter to zip() calls in test_adamuon.py

Co-authored-by: njzjz <9496702+njzjz@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: njzjz <9496702+njzjz@users.noreply.github.com>

Author: Copilot

source/tests/pt/test_adamuon.py

@@ -97,7 +97,9 @@ def test_optimizer_step_smoke(self) -> None:
         optimizer.step()
 
         # Verify all parameters with gradients changed
-        for i, (p, init_p) in enumerate(zip(model.parameters(), initial_params)):
+        for i, (p, init_p) in enumerate(
+            zip(model.parameters(), initial_params, strict=True)
+        ):
             if p.grad is not None:
                 self.assertFalse(
                     torch.allclose(p, init_p),
@@ -369,7 +371,7 @@ def test_state_dict_save_load(self) -> None:
         params1 = list(optimizer.param_groups[0]["params"])
         params2 = list(optimizer2.param_groups[0]["params"])
 
-        for p1, p2 in zip(params1, params2):
+        for p1, p2 in zip(params1, params2, strict=True):
             s1 = optimizer.state[p1]
             s2 = optimizer2.state[p2]
             self.assertEqual(set(s1.keys()), set(s2.keys()))