Update test_loss_default_pf.py

iProzd · iProzd · commit f4fda39a41cc · 2026-05-13T18:25:43.000+08:00
diff --git a/source/tests/pt/test_loss_default_pf.py b/source/tests/pt/test_loss_default_pf.py
@@ -298,7 +298,7 @@ def test_serialize_deserialize(self) -> None:
         )
         data = loss_fn.serialize()
         self.assertTrue(data["use_default_pf"])
-        self.assertEqual(data["@version"], 3)
+        self.assertEqual(data["@version"], 4)
 
         loss_fn2 = EnergyStdLoss.deserialize(data)
         self.assertTrue(loss_fn2.use_default_pf)

Original file line number	Diff line number	Diff line change
`@@ -298,7 +298,7 @@ def test_serialize_deserialize(self) -> None:`
`298`	`298`	`)`
`299`	`299`	`data = loss_fn.serialize()`
`300`	`300`	`self.assertTrue(data["use_default_pf"])`
`301`		`- self.assertEqual(data["@version"], 3)`
	`301`	`+ self.assertEqual(data["@version"], 4)`
`302`	`302`
`303`	`303`	`loss_fn2 = EnergyStdLoss.deserialize(data)`
`304`	`304`	`self.assertTrue(loss_fn2.use_default_pf)`