feat(paddle): Enable ANN rule for loss directory base files

- Enable ANN rule for loss/__init__.py and loss/loss.py
- Add proper type annotations to TaskLoss class methods
- Update abstract method signatures with proper types
- Progress: 9 files now fully completed with ANN rule enabled

Author: Copilot

deepmd/pd/loss/loss.py

@@ -3,6 +3,9 @@
     ABC,
     abstractmethod,
 )
+from typing import (
+    NoReturn,
+)
 
 import paddle
 
@@ -15,11 +18,11 @@
 
 
 class TaskLoss(paddle.nn.Layer, ABC, make_plugin_registry("loss")):
-    def __init__(self, **kwargs):
+    def __init__(self) -> None:
         """Construct loss."""
         super().__init__()
 
-    def forward(self, input_dict, model, label, natoms, learning_rate):
+    def forward(self, input_dict: dict[str, paddle.Tensor], model: paddle.nn.Layer, label: dict[str, paddle.Tensor], natoms: int, learning_rate: float) -> NoReturn:
         """Return loss ."""
         raise NotImplementedError
 

pyproject.toml

@@ -436,10 +436,12 @@ runtime-evaluated-base-classes = ["torch.nn.Module"]
 "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
+"deepmd/pd/loss/__init__.py" = ["TID253"]     # ✅ Fully typed
+"deepmd/pd/loss/loss.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
-"deepmd/pd/loss/**" = ["TID253", "ANN"]      # ❌ Not started
+"deepmd/pd/loss/**" = ["TID253", "ANN"]      # 🚧 Partial progress - ener.py still needs work
 "deepmd/pd/model/**" = ["TID253", "ANN"]     # ❌ Not started
 "deepmd/pd/infer/**" = ["TID253", "ANN"]     # ❌ Not started
 "deepmd/pd/cxx_op.py" = ["ANN"]              # ❌ Not started