fix UT

deepmd/dpmodel/fitting/property_fitting.py

@@ -75,6 +75,7 @@ def __init__(
         trainable: Union[bool, list[bool]] = True,
         intensive: bool = False,
         bias_method: str = "normal",
+        property_name: Union[str, list] = "property",
         resnet_dt: bool = True,
         numb_fparam: int = 0,
         numb_aparam: int = 0,
@@ -90,6 +91,7 @@ def __init__(
         self.task_dim = task_dim
         self.intensive = intensive
         self.bias_method = bias_method
+        self.property_name = property_name
         super().__init__(
             var_name="property",
             ntypes=ntypes,

deepmd/dpmodel/output_def.py

@@ -3,6 +3,7 @@
 from enum import (
     IntEnum,
 )
+from typing import Union
 
 
 def check_shape(
@@ -197,7 +198,7 @@ def __init__(
         r_hessian: bool = False,
         magnetic: bool = False,
         intensive: bool = False,
-        sub_var_name: list[str] = [],
+        sub_var_name: Union[list[str], str, None] = None,
     ) -> None:
         self.name = name
         self.shape = list(shape)
@@ -221,6 +222,8 @@ def __init__(
         self.r_hessian = r_hessian
         self.magnetic = magnetic
         self.intensive = intensive
+        if isinstance(sub_var_name, str):
+            sub_var_name = [sub_var_name]
         self.sub_var_name = sub_var_name
         if self.r_hessian:
             if not self.reducible: