fix ut

Author: iProzd

deepmd/pd/loss/ener.py

@@ -592,7 +592,7 @@ def deserialize(cls, data: dict) -> "TaskLoss":
             The deserialized loss module
         """
         data = data.copy()
-        check_version_compatibility(data.pop("@version"), 2, 1)
+        check_version_compatibility(data.pop("@version"), 3, 1)
         data.pop("@class")
         return cls(**data)
 

deepmd/tf/loss/ener.py

@@ -571,7 +571,7 @@ def deserialize(cls, data: dict, suffix: str = "") -> "Loss":
             The deserialized loss module
         """
         data = data.copy()
-        check_version_compatibility(data.pop("@version"), 2, 1)
+        check_version_compatibility(data.pop("@version"), 3, 1)
         data.pop("@class")
         return cls(**data)