feat(paddle): Enable ANN rule for additional utility files and complete type annotations

- Enable ANN rule for 5 more utility files: learning_rate.py, dp_random.py, update_sel.py, preprocess.py, spin.py
- Add missing type annotations to preprocess.py and spin.py functions
- Update gradual enablement tracking: now 7 files fully completed with ANN rule enabled

Author: Copilot

deepmd/pd/utils/preprocess.py

@@ -6,7 +6,7 @@
 log = logging.getLogger(__name__)
 
 
-def compute_smooth_weight(distance, rmin: float, rmax: float):
+def compute_smooth_weight(distance: paddle.Tensor, rmin: float, rmax: float) -> paddle.Tensor:
     """Compute smooth weight for descriptor elements."""
     if rmin >= rmax:
         raise ValueError("rmin should be less than rmax.")
@@ -17,7 +17,7 @@ def compute_smooth_weight(distance, rmin: float, rmax: float):
     return vv
 
 
-def compute_exp_sw(distance, rmin: float, rmax: float):
+def compute_exp_sw(distance: paddle.Tensor, rmin: float, rmax: float) -> paddle.Tensor:
     """Compute the exponential switch function for neighbor update."""
     if rmin >= rmax:
         raise ValueError("rmin should be less than rmax.")

deepmd/pd/utils/spin.py

@@ -4,10 +4,10 @@
 
 
 def concat_switch_virtual(
-    extended_tensor,
-    extended_tensor_virtual,
+    extended_tensor: paddle.Tensor,
+    extended_tensor_virtual: paddle.Tensor,
     nloc: int,
-):
+) -> paddle.Tensor:
     """
     Concat real and virtual extended tensors, and switch all the local ones to the first nloc * 2 atoms.
     - [:, :nloc]: original nloc real atoms.

pyproject.toml

@@ -431,6 +431,11 @@ runtime-evaluated-base-classes = ["torch.nn.Module"]
 # Completed files with full type annotations:
 "deepmd/pd/entrypoints/main.py" = ["TID253"]  # ✅ Fully typed
 "deepmd/pd/train/wrapper.py" = ["TID253"]     # ✅ Fully typed
+"deepmd/pd/utils/learning_rate.py" = ["TID253"]  # ✅ Fully typed
+"deepmd/pd/utils/dp_random.py" = ["TID253"]   # ✅ Fully typed
+"deepmd/pd/utils/update_sel.py" = ["TID253"]  # ✅ Fully typed
+"deepmd/pd/utils/preprocess.py" = ["TID253"]  # ✅ Fully typed
+"deepmd/pd/utils/spin.py" = ["TID253"]        # ✅ Fully typed
 # TODO: Complete type hints and remove ANN exclusion for remaining files:
 "deepmd/pd/train/**" = ["TID253", "ANN"]     # 🚧 Partial progress - training.py still needs work
 "deepmd/pd/utils/**" = ["TID253", "ANN"]     # 🚧 Partial progress - utils.py partially done