[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/utils/neighbor_graph/angles.py

@@ -6,9 +6,13 @@
 center_edge_pairs; a_rcut filters the participating edges.
 """
 
-from __future__ import annotations
+from __future__ import (
+    annotations,
+)
 
-from typing import TYPE_CHECKING
+from typing import (
+    TYPE_CHECKING,
+)
 
 import array_api_compat
 
@@ -17,9 +21,17 @@
 
 import dataclasses
 
-from .graph import GraphLayout, NeighborGraph, pad_and_guard_angles
-from .pairs import center_edge_pairs
-from .segment import segment_sum
+from .graph import (
+    GraphLayout,
+    NeighborGraph,
+    pad_and_guard_angles,
+)
+from .pairs import (
+    center_edge_pairs,
+)
+from .segment import (
+    segment_sum,
+)
 
 
 def build_angle_index(

deepmd/dpmodel/utils/neighbor_graph/pairs.py

@@ -24,7 +24,9 @@
 ``E ~ N * nnei`` it costs ``O(N**2 * nnei**2)`` memory.
 """
 
-from __future__ import annotations
+from __future__ import (
+    annotations,
+)
 
 from typing import (
     Any,

source/tests/common/dpmodel/test_angle_builder.py

@@ -1,3 +1,4 @@
+# SPDX-License-Identifier: LGPL-3.0-or-later
 import numpy as np
 import pytest
 

source/tests/common/dpmodel/test_graph_angle_cos_parity.py

@@ -1,7 +1,9 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 """Parity tests for graph_angle_cos vs dpa3 dense cosine_ij and se_t dot form."""
 
-from __future__ import annotations
+from __future__ import (
+    annotations,
+)
 
 import numpy as np
 
@@ -11,7 +13,6 @@
     graph_angle_cos,
 )
 
-
 # ---------------------------------------------------------------------------
 # Step 1/4: explicit geometry sanity
 # ---------------------------------------------------------------------------