[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

deepmd/dpmodel/output_def.py

@@ -267,7 +267,7 @@ def __getitem__(
     def get_data(self) -> dict[str, OutputVariableDef]:
         return self.var_defs
 
-    def keys(self):  # noqa: ANN201
+    def keys(self):
         return self.var_defs.keys()
 
 
@@ -319,25 +319,25 @@ def get_data(
     ) -> dict[str, OutputVariableDef]:
         return self.var_defs
 
-    def keys(self):  # noqa: ANN201
+    def keys(self):
         return self.var_defs.keys()
 
-    def keys_outp(self):  # noqa: ANN201
+    def keys_outp(self):
         return self.def_outp.keys()
 
-    def keys_redu(self):  # noqa: ANN201
+    def keys_redu(self):
         return self.def_redu.keys()
 
-    def keys_derv_r(self):  # noqa: ANN201
+    def keys_derv_r(self):
         return self.def_derv_r.keys()
 
-    def keys_hess_r(self):  # noqa: ANN201
+    def keys_hess_r(self):
         return self.def_hess_r.keys()
 
-    def keys_derv_c(self):  # noqa: ANN201
+    def keys_derv_c(self):
         return self.def_derv_c.keys()
 
-    def keys_derv_c_redu(self):  # noqa: ANN201
+    def keys_derv_c_redu(self):
         return self.def_derv_c_redu.keys()
 
 

deepmd/dpmodel/utils/network.py

@@ -38,7 +38,7 @@
 )
 
 
-def sigmoid_t(x):  # noqa: ANN001, ANN201
+def sigmoid_t(x):
     """Sigmoid."""
     if array_api_compat.is_jax_array(x):
         from deepmd.jax.env import (
@@ -55,7 +55,7 @@ class Identity(NativeOP):
     def __init__(self) -> None:
         super().__init__()
 
-    def call(self, x):  # noqa: ANN001, ANN201
+    def call(self, x):
         """The Identity operation layer."""
         return x
 
@@ -260,7 +260,7 @@ def dim_out(self) -> int:
         return self.w.shape[1]
 
     @support_array_api(version="2022.12")
-    def call(self, x):  # noqa: ANN001, ANN201
+    def call(self, x):
         """Forward pass.
 
         Parameters
@@ -301,22 +301,22 @@ def get_activation_fn(activation_function: str) -> Callable[[np.ndarray], np.nda
     activation_function = activation_function.lower()
     if activation_function == "tanh":
 
-        def fn(x):  # noqa: ANN001, ANN202  # noqa: ANN001, ANN202
+        def fn(x):
             xp = array_api_compat.array_namespace(x)
             return xp.tanh(x)
 
         return fn
     elif activation_function == "relu":
 
-        def fn(x):  # noqa: ANN001, ANN202
+        def fn(x):
             xp = array_api_compat.array_namespace(x)
             # https://stackoverflow.com/a/47936476/9567349
             return x * xp.astype(x > 0, x.dtype)
 
         return fn
     elif activation_function in ("gelu", "gelu_tf"):
 
-        def fn(x):  # noqa: ANN001, ANN202
+        def fn(x):
             xp = array_api_compat.array_namespace(x)
             # generated by GitHub Copilot
             return (
@@ -328,7 +328,7 @@ def fn(x):  # noqa: ANN001, ANN202
         return fn
     elif activation_function == "relu6":
 
-        def fn(x):  # noqa: ANN001, ANN202
+        def fn(x):
             xp = array_api_compat.array_namespace(x)
             # generated by GitHub Copilot
             return xp.where(
@@ -338,22 +338,22 @@ def fn(x):  # noqa: ANN001, ANN202
         return fn
     elif activation_function == "softplus":
 
-        def fn(x):  # noqa: ANN001, ANN202
+        def fn(x):
             xp = array_api_compat.array_namespace(x)
             # generated by GitHub Copilot
             return xp.log(1 + xp.exp(x))
 
         return fn
     elif activation_function == "sigmoid":
 
-        def fn(x):  # noqa: ANN001, ANN202
+        def fn(x):
             # generated by GitHub Copilot
             return sigmoid_t(x)
 
         return fn
     elif activation_function == "silu":
 
-        def fn(x):  # noqa: ANN001, ANN202
+        def fn(x):
             # generated by GitHub Copilot
             return x * sigmoid_t(x)
 
@@ -362,13 +362,13 @@ def fn(x):  # noqa: ANN001, ANN202
         "custom_silu"
     ):
 
-        def sigmoid(x):  # noqa: ANN001, ANN202
+        def sigmoid(x):
             return 1 / (1 + np.exp(-x))
 
-        def silu(x):  # noqa: ANN001, ANN202
+        def silu(x):
             return x * sigmoid(x)
 
-        def silu_grad(x):  # noqa: ANN001, ANN202
+        def silu_grad(x):
             sig = sigmoid(x)
             return sig + x * sig * (1 - sig)
 
@@ -380,7 +380,7 @@ def silu_grad(x):  # noqa: ANN001, ANN202
         slope = float(silu_grad(threshold))
         const = float(silu(threshold))
 
-        def fn(x):  # noqa: ANN001, ANN202
+        def fn(x):
             xp = array_api_compat.array_namespace(x)
             return xp.where(
                 x < threshold,
@@ -391,7 +391,7 @@ def fn(x):  # noqa: ANN001, ANN202
         return fn
     elif activation_function.lower() in ("none", "linear"):
 
-        def fn(x):  # noqa: ANN001, ANN202
+        def fn(x):
             return x
 
         return fn
@@ -535,7 +535,7 @@ def __getitem__(self, key: str) -> Any:
     def dim_out(self) -> int:
         return self.w.shape[0]
 
-    def call(self, x):  # noqa: ANN001, ANN201
+    def call(self, x):
         """Forward pass.
 
         Parameters
@@ -552,11 +552,11 @@ def call(self, x):  # noqa: ANN001, ANN201
         return y
 
     @staticmethod
-    def layer_norm_numpy(  # noqa: ANN205
-        x,  # noqa: ANN001
+    def layer_norm_numpy(
+        x,
         shape: tuple[int, ...],
-        weight=None,  # noqa: ANN001
-        bias=None,  # noqa: ANN001
+        weight=None,
+        bias=None,
         eps: float = 1e-5,
     ):
         xp = array_api_compat.array_namespace(x)
@@ -633,7 +633,7 @@ def check_shape_consistency(self) -> None:
                         f"output {self.layers[ii].dim_out}",
                     )
 
-        def call(self, x):  # noqa: ANN001, ANN202
+        def call(self, x):
             """Forward pass.
 
             Parameters
@@ -650,7 +650,7 @@ def call(self, x):  # noqa: ANN001, ANN202
                 x = layer(x)
             return x
 
-        def call_until_last(self, x):  # noqa: ANN001, ANN202
+        def call_until_last(self, x):
             """Return the output before last layer.
 
             Parameters
@@ -1025,9 +1025,9 @@ def deserialize(cls, data: dict) -> "NetworkCollection":
         return cls(**data)
 
 
-def aggregate(  # noqa: ANN201
-    data,  # noqa: ANN001
-    owners,  # noqa: ANN001
+def aggregate(
+    data,
+    owners,
     average: bool = True,
     num_owner: Optional[int] = None,
 ):
@@ -1065,10 +1065,10 @@ def aggregate(  # noqa: ANN201
     return output
 
 
-def get_graph_index(  # noqa: ANN201
-    nlist,  # noqa: ANN001
-    nlist_mask,  # noqa: ANN001
-    a_nlist_mask,  # noqa: ANN001
+def get_graph_index(
+    nlist,
+    nlist_mask,
+    a_nlist_mask,
     nall: int,
     use_loc_mapping: bool = True,
 ):