Merge remote-tracking branch 'origin/feat-preset-bias' into feat-pres…

…et-bias
wanghan-iapcm · Sep 11, 2024 · 836e6c2 · 836e6c2
2 parents 3297020 + f2885d5
commit 836e6c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/model/model/__init__.py b/deepmd/pt/model/model/__init__.py
@@ -153,7 +153,7 @@ def get_zbl_model(model_params):
 
 def _convert_preset_out_bias_to_array(preset_out_bias, type_map):
     if preset_out_bias is not None:
-        for kk in preset_out_bias.keys():
+        for kk in preset_out_bias:
             if len(preset_out_bias[kk]) != len(type_map):
                 raise ValueError(
                     "length of the preset_out_bias should be the same as the type_map"