fix

Author: HydrogenSulfate

deepmd/pd/model/atomic_model/dp_atomic_model.py

@@ -3,6 +3,7 @@
 import logging
 from typing import (
     Optional,
+    Union,
 )
 
 import paddle
@@ -67,7 +68,7 @@ def __init__(
         self.eval_fitting_last_layer_list = []
 
         # register 'type_map' as buffer
-        def _string_to_array(s: str | list[str]) -> list[int]:
+        def _string_to_array(s: Union[str, list[str]]) -> list[int]:
             return [ord(c) for c in s]
 
         self.register_buffer(

source/tests/pd/model/test_descriptor_dpa1.py

@@ -368,7 +368,7 @@ def translate_se_atten_and_type_embd_dicts_to_dpa1(
     source_dict,
     type_embd_dict,
 ):
-    all_keys = list(target_dict.keys())
+    all_keys = [key for key in target_dict.keys() if "buffer_" not in key]
     record = [False for ii in all_keys]
     for kk, vv in source_dict.items():
         tk = "se_atten." + kk

source/tests/pd/model/test_descriptor_dpa2.py

@@ -190,7 +190,7 @@ def translate_type_embd_dicts_to_dpa2(
     source_dict,
     type_embd_dict,
 ):
-    all_keys = list(target_dict.keys())
+    all_keys = [key for key in target_dict.keys() if "buffer_" not in key]
     record = [False for ii in all_keys]
     for kk, vv in source_dict.items():
         record[all_keys.index(kk)] = True