do not save none

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · May 28, 2024 · 0bcd410 · 0bcd410
1 parent cedaca8
commit 0bcd410
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/deepmd/pt/utils/serialization.py b/deepmd/pt/utils/serialization.py
@@ -50,10 +50,10 @@ def serialize_from_file(model_file: str) -> dict:
         "pt_version": torch.__version__,
         "model": model_dict,
         "model_def_script": model_def_script,
-        "@variables": {
-            "min_nbor_dist": model.get_min_nbor_dist(),
-        },
+        "@variables": {},
     }
+    if model.get_min_nbor_dist() is not None:
+        data["@variables"]["min_nbor_dist"] = model.get_min_nbor_dist()
     return data