solve serialize UT error

Chengqian-Zhang · Aug 27, 2024 · 3fd5697 · 3fd5697
1 parent a865f23
commit 3fd5697
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deepmd/dpmodel/fitting/property_fitting.py b/deepmd/dpmodel/fitting/property_fitting.py
@@ -31,6 +31,7 @@ def __init__(
         rcond: Optional[float] = None,
         trainable: Union[bool, List[bool]] = True,
         intensive: bool = False,
+        bias_method: str = "normal",
         resnet_dt: bool = True,
         numb_fparam: int = 0,
         numb_aparam: int = 0,
@@ -44,6 +45,7 @@ def __init__(
     ):
         self.task_dim = task_dim
         self.intensive = intensive
+        self.bias_method = bias_method
         super().__init__(
             var_name="property",
             ntypes=ntypes,