fix: UTs

anyangml · Feb 22, 2024 · 6da24c0 · 6da24c0
1 parent eee3048
commit 6da24c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/model/task/fitting.py b/deepmd/pt/model/task/fitting.py
@@ -428,7 +428,7 @@ def serialize(self) -> dict:
             ## NOTICE:  not supported by far
             "tot_ener_zero": False,
             "trainable": [True] * (len(self.neuron) + 1),
-            "atom_ener": [],
+            "atom_ener": None,
             "layer_name": None,
             "use_aparam_as_mask": False,
             "spin": None,