fix: fix YAML conversion

Fix deepmodeling#4564. Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Jan 20, 2025 · e033f4f · e033f4f
1 parent a3242c4
commit e033f4f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepmd/dpmodel/utils/serialization.py b/deepmd/dpmodel/utils/serialization.py
@@ -113,7 +113,9 @@ def save_dp_model(filename: str, model_dict: dict) -> None:
                 "@version": 1,
                 "dtype": x.dtype.name,
                 "value": x.tolist(),
-            },
+            }
+            if isinstance(x, np.ndarray)
+            else x,
         )
         with open(filename, "w") as f:
             yaml.safe_dump(