fix bugs

backend/dynamic_metadata.py

@@ -36,7 +36,7 @@ def dynamic_metadata(
     settings: Optional[Dict[str, object]] = None,
 ):
     assert field in ["optional-dependencies", "entry-points", "scripts"]
-    _, _, find_libpython_requires, extra_scripts, tf_version, pt_version = (
+    _, _, find_libpython_requires, extra_scripts, tf_version, pt_version, pd_version = (
         get_argument_from_env()
     )
     with Path("pyproject.toml").open("rb") as f:

backend/read_env.py

@@ -153,6 +153,6 @@ def get_argument_from_env() -> Tuple[str, list, list, dict, str, str]:
 
 def set_scikit_build_env():
     """Set scikit-build environment variables before executing scikit-build."""
-    cmake_minimum_required_version, cmake_args, _, _, _, _ = get_argument_from_env()
+    cmake_minimum_required_version, cmake_args, _, _, _, _, _ = get_argument_from_env()
     os.environ["SKBUILD_CMAKE_MINIMUM_VERSION"] = cmake_minimum_required_version
     os.environ["SKBUILD_CMAKE_ARGS"] = ";".join(cmake_args)

deepmd/pd/model/descriptor/se_atten.py

@@ -484,9 +484,9 @@ def forward(
         )
         # nb x nloc x nnei
         exclude_mask = self.emask(nlist, extended_atype)
-        nlist = paddle.where(exclude_mask != 0, nlist, -1)
+        nlist = paddle.where(exclude_mask != 0, nlist, paddle.full_like(nlist, -1))
         nlist_mask = nlist != -1
-        nlist = paddle.where(nlist == -1, 0, nlist)
+        nlist = paddle.where(nlist == -1, paddle.zeros_like(nlist), nlist)
         sw = paddle.squeeze(sw, -1)
         # nf x nloc x nt -> nf x nloc x nnei x nt
         atype_tebd = extended_atype_embd[:, :nloc, :]