tiny fix

Former-commit-id: 38b6b0f52e
2025-12-14 10:56:56 +08:00 · 2024-06-16 01:06:41 +08:00
parent 96b82ccd4d
commit c0c6b8075a
22 changed files with 27 additions and 25 deletions
--- a/tests/model/test_base.py
+++ b/tests/model/test_base.py
@@ -41,7 +41,7 @@ def compare_model(model_a: "torch.nn.Module", model_b: "torch.nn.Module"):
    state_dict_b = model_b.state_dict()
    assert set(state_dict_a.keys()) == set(state_dict_b.keys())
    for name in state_dict_a.keys():
-        assert torch.allclose(state_dict_a[name], state_dict_b[name]) is True
+        assert torch.allclose(state_dict_a[name], state_dict_b[name])


@pytest.fixture