param list test polist

2021-06-22 21:35:56 +08:00 · 2021-06-22 21:35:56 +08:00 · ca9b94f590
parent 698fc6fe88
commit ca9b94f590
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/test/test_param_list.py
+++ b/python/jittor/test/test_param_list.py
@ -25,7 +25,7 @@ class TestParamList(unittest.TestCase):
                    "bbb": jt.rand(10)
                })
        net = Net()
-        assert net.state_dict().keys() == ['ps1.0', 'ps1.1', 'ps2.aaa', 'ps2.bbb']
+        assert list(net.state_dict().keys()) == ['ps1.0', 'ps1.1', 'ps2.aaa', 'ps2.bbb']

 if __name__ == "__main__":
    unittest.main()